| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_reduction.cpp | 22 array<ptrdiff_t, 0> reduction_axis; in test_trivial_reductions() local
31 array<ptrdiff_t, 0> reduction_axis; in test_trivial_reductions() local
43 array<ptrdiff_t, 0> reduction_axis; in test_trivial_reductions() local
275 array<ptrdiff_t, 2> reduction_axis; in test_full_reductions() local
317 array<ptrdiff_t, 1> reduction_axis; in test_user_defined_reductions() local
343 array<ptrdiff_t, 2> reduction_axis; in test_tensor_maps() local
374 array<int, 2> reduction_axis; in test_static_dims() local
378 Eigen::IndexList<Eigen::type2index<1>, Eigen::type2index<3> > reduction_axis; in test_static_dims() local
404 array<int, 2> reduction_axis; in test_innermost_last_dims() local
409 Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<1> > reduction_axis; in test_innermost_last_dims() local
[all …]
|
| H A D | cxx11_tensor_index_list.cpp | 21 constexpr auto reduction_axis = make_index_list(0, 1, 2); in test_static_index_list() local
275 auto reduction_axis = make_index_list(dim1, dim2, dim3); in test_dynamic_index_list() local
306 auto reduction_axis = make_index_list(0, dim2, 2, dim4); in test_mixed_index_list() local
|
| H A D | cxx11_tensor_gpu.cu | 231 array<Eigen::DenseIndex, 2> reduction_axis; in test_gpu_reduction() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLReduceMean.cpp | 41 Status validate_config(const ITensorInfo *input, const Coordinates &reduction_axis, bool keep_dims,… in validate_config()
109 void CLReduceMean::configure(ICLTensor *input, const Coordinates &reduction_axis, bool keep_dims, I… in configure()
114 …(const CLCompileContext &compile_context, ICLTensor *input, const Coordinates &reduction_axis, boo… in configure()
198 Status CLReduceMean::validate(const ITensorInfo *input, const Coordinates &reduction_axis, bool kee… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEReduceMean.cpp | 37 Status validate_config(const ITensorInfo *input, const Coordinates &reduction_axis, bool keep_dims,… in validate_config()
98 Status NEReduceMean::validate(const ITensorInfo *input, const Coordinates &reduction_axis, bool kee… in validate()
103 void NEReduceMean::configure(ITensor *input, const Coordinates &reduction_axis, bool keep_dims, ITe… in configure()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | adjust_contrast_op.h | 45 reduction_axis; in operator() local
93 reduction_axis; in operator() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | batch_norm_op.h | 89 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator() local
|
| H A D | redux_functor.h | 264 Eigen::array<Eigen::Index, 1> reduction_axis = {0}; in operator() local
|
| H A D | fused_batch_norm_op.cu.cc | 71 Eigen::IndexList<Eigen::type2index<0> > reduction_axis; in operator ()() local
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/ |
| H A D | ShapeCalculator.h | 51 …TensorShape calculate_reduce_mean_shape(ITensorInfo *input, const Coordinates &reduction_axis, boo… in calculate_reduce_mean_shape()
|