Home
last modified time | relevance | path

Searched defs:dim_list (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/util/test/
H A Dreduce_test.cpp40 const optional<ArrayRef<int64_t>>& dim_list) { in _apply_over_dim_list()
53 optional<ArrayRef<int64_t>> dim_list; in TEST() local
144 optional<ArrayRef<int64_t>> dim_list; in TEST() local
164 optional<ArrayRef<int64_t>> dim_list; in TEST() local
241 optional<ArrayRef<int64_t>> dim_list; in TEST() local
354 optional<ArrayRef<int64_t>> dim_list; in TEST() local
431 optional<ArrayRef<int64_t>> dim_list; in TEST() local
469 optional<ArrayRef<int64_t>> dim_list = in TEST() local
479 optional<ArrayRef<int64_t>> dim_list; in TEST() local
516 optional<ArrayRef<int64_t>> dim_list; in TEST() local
[all …]
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dreduce_util.cpp37 const exec_aten::optional<exec_aten::ArrayRef<int64_t>>& dim_list) { in check_dim_list_is_valid()
67 const exec_aten::ArrayRef<int64_t>& dim_list) { in check_dim_in_dim_list()
102 const exec_aten::optional<exec_aten::ArrayRef<int64_t>>& dim_list) { in get_reduced_dim_product()
152 const exec_aten::optional<exec_aten::ArrayRef<int64_t>>& dim_list) { in get_out_numel()
201 const exec_aten::optional<exec_aten::ArrayRef<int64_t>>& dim_list, in get_init_index()
261 const exec_aten::optional<exec_aten::ArrayRef<int64_t>>& dim_list, in compute_reduced_out_size()
313 const exec_aten::optional<exec_aten::ArrayRef<int64_t>>& dim_list, in resize_reduction_out()
331 const optional<ArrayRef<int64_t>>& dim_list, in check_reduction_args()
381 optional<ArrayRef<int64_t>> dim_list, in check_mean_dim_args()
401 ArrayRef<int64_t> dim_list, in check_amin_amax_args()
H A Dreduce_util.h460 const exec_aten::optional<exec_aten::ArrayRef<int64_t>>& dim_list, in map_reduce_over_dim_list()
546 const exec_aten::optional<exec_aten::ArrayRef<int64_t>>& dim_list, in reduce_over_dim_list()
580 const exec_aten::optional<exec_aten::ArrayRef<int64_t>>& dim_list, in compute_reduced_out_dim()
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_amin_test.cpp57 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims_1, 1}}; in test_amin_out_invalid_dimensions() local
90 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims_1, 1}}; in test_amin_out_invalid_shape() local
120 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims_1, 1}}; in test_amin_out_dtype() local
274 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims_1, 1}}; in TEST_F() local
305 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims, 1}}; in TEST_F() local
H A Dop_amax_test.cpp57 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims_1, 1}}; in test_amax_out_invalid_dimensions() local
90 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims_1, 1}}; in test_amax_out_invalid_shape() local
120 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims_1, 1}}; in test_amax_out_dtype() local
274 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims_1, 1}}; in TEST_F() local
305 ArrayRef<int64_t> dim_list{ArrayRef<int64_t>{dims, 1}}; in TEST_F() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_var.cpp25 optional<ArrayRef<int64_t>> dim_list, in compute_variance()
62 optional<ArrayRef<int64_t>> dim_list, in var_out()
105 optional<ArrayRef<int64_t>> dim_list, in var_correction_out()
H A Dop_amax.cpp25 ArrayRef<int64_t> dim_list, in amax_out()
H A Dop_amin.cpp25 ArrayRef<int64_t> dim_list, in amin_out()
H A Dop_sum.cpp23 optional<ArrayRef<int64_t>> dim_list, in sum_dim_out()
H A Dop_mean.cpp24 optional<ArrayRef<int64_t>> dim_list, in mean_dim_out()
H A Dop_any.cpp52 optional<ArrayRef<int64_t>> dim_list, in any_dims_out()
/aosp_15_r20/external/executorch/backends/cadence/hifi/operators/
H A Dop_mean.cpp31 optional<ArrayRef<int64_t>> dim_list, in prepare_data()
62 optional<ArrayRef<int64_t>> dim_list, in mean_dim_out()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DFlip.cpp13 Tensor flip(const at::Tensor& self, const IntArrayRef dim_list) { in flip()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DFlip.cpp49 const std::vector<int64_t>& dim_list, in add_flip_node()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Dsmall_constant_optimization.cc139 std::vector<int64_t> dim_list; in ExtractSmallTensorValue() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dreduce_join_op.cc50 int64_t output_index, const gtl::InlinedVector<int32, 8>& dim_list, in LinearSubIndexToFullIndex()
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp2511 void test_flip(const at::IntArrayRef input_shape, const at::IntArrayRef dim_list) { in test_flip()
3468 void test_mean_dim(const at::IntArrayRef input_shape, const at::IntArrayRef dim_list, bool keepdim=… in test_mean_dim()
4895 void test_sum_dim(const at::IntArrayRef input_shape, const at::IntArrayRef dim_list, bool keepdim=f… in test_sum_dim()
5646 void test_var(const at::IntArrayRef input_shape, const at::IntArrayRef dim_list, bool unbiased=true… in test_var()