Home
last modified time | relevance | path

Searched defs:dim_order (Results 1 – 25 of 45) sorted by relevance

12

/aosp_15_r20/external/executorch/runtime/core/exec_aten/util/
H A Ddim_order_util.h22 bool validate_dim_order(const DimOrderType* dim_order, const size_t dims) { in validate_dim_order()
41 const DimOrderType* dim_order, in is_contiguous_dim_order()
60 const DimOrderType* dim_order, in is_channels_last_dim_order()
113 const DimOrderType* dim_order, in dim_order_to_stride_nocheck()
139 const DimOrderType* dim_order, in dim_order_to_stride()
161 DimOrderType dim_order; member
232 DimOrderType* dim_order) { in stride_to_dim_order()
H A Dtensor_util_aten.cpp37 exec_aten::DimOrderType dim_order[kTensorDimensionLimit]; in tensor_has_valid_dim_order() local
57 exec_aten::DimOrderType dim_order[kTensorDimensionLimit]; in tensor_is_default_or_channels_last_dim_order() local
/aosp_15_r20/external/executorch/kernels/test/
H A Dop__to_dim_order_copy_test.cpp58 exec_aten::optional<ArrayRef<int64_t>> dim_order, in op__to_dim_order_copy_out()
110 ArrayRef<int64_t> dim_order(dim_order_vec.data(), dim_order_vec.size()); in test_runner_static_cast() local
143 ArrayRef<int64_t> dim_order(dim_order_vec.data(), dim_order_vec.size()); in test_runner_to_bool() local
177 ArrayRef<int64_t> dim_order(dim_order_vec.data(), dim_order_vec.size()); in test_runner_from_bool() local
243 ArrayRef<int64_t> dim_order(dim_order_vec.data(), dim_order_vec.size()); in test_dynamic_shape() local
282 ArrayRef<int64_t> dim_order(dim_order_vec.data(), dim_order_vec.size()); in test_runner_hardcode_data() local
461 ArrayRef<int64_t> dim_order(dim_order_vec.data(), dim_order_vec.size()); in TEST_F() local
493 ArrayRef<int64_t> dim_order(dim_order_vec.data(), dim_order_vec.size()); in TEST_F() local
517 ArrayRef<int64_t> dim_order(dim_order_vec.data(), dim_order_vec.size()); in TEST_F() local
572 exec_aten::ArrayRef<int64_t> dim_order( in TEST_F() local
[all …]
/aosp_15_r20/external/executorch/runtime/core/exec_aten/testing_util/test/
H A Dtensor_factory_test.cpp131 uint8_t dim_order[dim] = {0, 1}; in TEST_F() local
160 uint8_t dim_order[dim] = {0, 1}; in TEST_F() local
190 uint8_t dim_order[dim] = {0, 1}; in TEST_F() local
323 uint8_t dim_order[dim] = {1, 0}; in TEST_F() local
360 uint8_t dim_order[dim] = {1, 0}; in TEST_F() local
397 uint8_t dim_order[dim] = {1, 0}; in TEST_F() local
435 uint8_t dim_order[dim] = {0, 10, 1, 3, 7, 8, 9, 2, 4, 6, 5}; in TEST_F() local
485 uint8_t dim_order[dim] = {0, 1, 2, 3, 4}; in TEST_F() local
517 uint8_t dim_order[dim] = {}; in TEST_F() local
631 uint8_t dim_order[dim] = {0, 1}; in TEST_F() local
[all …]
/aosp_15_r20/external/executorch/extension/aten_util/test/
H A Daten_bridge_test.cpp28 std::vector<Tensor::DimOrderType> dim_order(t.dim()); in get_default_dim_order() local
38 auto dim_order = get_default_dim_order(at_tensor); in TEST() local
58 auto dim_order = get_default_dim_order(at_tensor); in TEST() local
88 auto dim_order = get_default_dim_order(at_tensor); in TEST() local
118 auto dim_order = get_default_dim_order(at_tensor); in TEST() local
134 auto dim_order = get_default_dim_order(at_tensor); in TEST() local
/aosp_15_r20/external/executorch/runtime/core/portable_type/test/
H A Dtensor_impl_test.cpp41 DimOrderType dim_order[2] = {0, 1}; in TEST_F() local
65 DimOrderType dim_order[RANK] = {0, 1, 2, 3, 4}; in TEST_F() local
95 DimOrderType dim_order[3] = {0, 1, 2}; in TEST_F() local
124 DimOrderType dim_order[2] = {0, 1}; in TEST_F() local
152 DimOrderType dim_order[2] = {0, 1}; in TEST_F() local
199 DimOrderType dim_order[0] = {}; in TEST_F() local
225 DimOrderType dim_order[2] = {0, 1}; in TEST_F() local
362 DimOrderType dim_order[1] = {0}; in TEST_F() local
H A Dtensor_test.cpp46 TensorImpl::DimOrderType dim_order[1] = {0}; in TEST_F() local
57 TensorImpl::DimOrderType dim_order[2] = {0, 1}; in TEST_F() local
71 TensorImpl::DimOrderType dim_order[2] = {0}; in TEST_F() local
/aosp_15_r20/external/executorch/extension/tensor/
H A Dtensor_ptr.cpp31 std::vector<exec_aten::DimOrderType> dim_order; member
60 std::vector<exec_aten::DimOrderType> dim_order, in make_tensor_ptr()
133 std::vector<exec_aten::DimOrderType> dim_order, in make_tensor_ptr()
156 std::vector<exec_aten::DimOrderType> dim_order{ in clone_tensor_ptr() local
189 std::vector<exec_aten::DimOrderType> dim_order{ in clone_tensor_ptr() local
H A Dtensor_impl_ptr.cpp38 std::shared_ptr<std::vector<exec_aten::DimOrderType>> dim_order; member
59 std::vector<exec_aten::DimOrderType> dim_order, in make_tensor_impl_ptr()
134 std::vector<exec_aten::DimOrderType> dim_order, in make_tensor_impl_ptr()
H A Dtensor_ptr_maker.h52 TensorPtrMaker&& dim_order( in dim_order() function
53 std::vector<executorch::aten::DimOrderType> dim_order) { in dim_order()
/aosp_15_r20/external/executorch/runtime/core/exec_aten/util/test/
H A Ddim_order_util_test.cpp208 exec_aten::DimOrderType dim_order[3] = {0, 0, 0}; in TEST() local
220 exec_aten::DimOrderType dim_order[4] = {0, 0, 0, 0}; in TEST() local
231 std::vector<exec_aten::DimOrderType> dim_order(i); in TEST() local
245 std::vector<exec_aten::DimOrderType> dim_order(i); in TEST() local
254 std::vector<exec_aten::DimOrderType> dim_order(i); in TEST() local
/aosp_15_r20/external/executorch/kernels/aten/cpu/
H A Dop__to_dim_order_copy.cpp28 Optional<MemoryFormat> get_memory_format(OptionalArrayRef<int64_t> dim_order) { in get_memory_format()
46 exec_aten::OptionalArrayRef<int64_t> dim_order, in check__to_dim_order_copy_args()
98 OptionalArrayRef<int64_t> dim_order, in _to_dim_order_copy_out()
116 OptionalArrayRef<int64_t> dim_order, in _to_dim_order_copy_out()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/
H A DTensor.cpp31 std::vector<int64_t> dim_order(ndim); in calculate_dim_order() local
52 const std::vector<int64_t>& dim_order) { in calculate_strides()
110 bool dim_order_is_valid(const std::vector<int64_t>& dim_order) { in dim_order_is_valid()
536 const std::vector<int64_t>& dim_order, in vTensor()
775 std::vector<int64_t>& dim_order, in transpose_dim_order_inplace()
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop__to_dim_order_copy.cpp39 DimOrderArrayRef dim_order = self.dim_order(); in coordinateToIndexWithDimOrder() local
84 OptionalArrayRef<int64_t> dim_order, in _to_dim_order_copy_out()
119 OptionalArrayRef<int64_t> dim_order, in _to_dim_order_copy_out()
/aosp_15_r20/external/executorch/runtime/core/exec_aten/testing_util/
H A Dtensor_factory.h129 const std::vector<uint8_t>& dim_order) { in check_dim_order()
152 const std::vector<uint8_t>& dim_order) { in strides_from_dim_order()
/aosp_15_r20/external/executorch/runtime/executor/
H A Dmethod_meta.cpp70 Span<const uint8_t> dim_order, in TensorInfo()
83 Span<const uint8_t> TensorInfo::dim_order() const { in dim_order() function in executorch::runtime::TensorInfo
H A Dtensor_parser_portable.cpp78 exec_aten::DimOrderType* dim_order = nullptr; in parseTensor() local
/aosp_15_r20/external/pytorch/torch/backends/_nnapi/
H A Dserializer.py292 def fix_shape(shape, dim_order): argument
309 def reverse_map_dim(dim_order, d): argument
374 def torch_tensor_to_operand(self, tensor, dim_order): argument
440 self, tensor, dim_order=DimOrder.UNKNOWN_CONSTANT argument
512 self, jitval, dim_order=DimOrder.PRESUMED_CONTIGUOUS argument
/aosp_15_r20/external/executorch/extension/tensor/test/
H A Dtensor_ptr_maker_test.cpp45 std::vector<exec_aten::DimOrderType> dim_order = {0, 1}; in TEST_F() local
67 std::vector<exec_aten::DimOrderType> dim_order = {0, 1}; in TEST_F() local
/aosp_15_r20/external/executorch/kernels/aten/cpu/util/
H A Dcopy_ops_util.cpp22 exec_aten::OptionalArrayRef<int64_t> dim_order, in check__to_dim_order_copy_args()
/aosp_15_r20/external/executorch/backends/arm/operators/
H A Dop_batch_norm.py30 def augment_shape_rank(self, shape, dim_order): argument
/aosp_15_r20/external/executorch/kernels/portable/test/
H A Dop_mul_test.cpp52 std::vector<exec_aten::DimOrderType> dim_order = {0, 1}; in TEST_F() local
/aosp_15_r20/external/executorch/runtime/core/portable_type/
H A Dtensor_impl.cpp48 DimOrderType* dim_order, in TensorImpl()
H A Dtensor.h104 const ArrayRef<DimOrderType> dim_order() const { in dim_order() function
/aosp_15_r20/external/executorch/examples/models/llama3_2_vision/cross_attention/
H A Dcross_attention_mask_test.cpp24 TensorImpl::DimOrderType dim_order[2] = {0, 1}; in TEST() local

12