| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor.h | 824 gtl::ArraySlice<int64_t> new_sizes, in FillDimsAndValidateCompatibleShape()
837 gtl::ArraySlice<int64_t> new_sizes, in FillDimsAndValidateCompatibleShape()
880 gtl::ArraySlice<int64_t> new_sizes) { in shaped()
889 gtl::ArraySlice<int64_t> new_sizes) { in bit_casted_shaped()
898 gtl::ArraySlice<int64_t> new_sizes) { in unaligned_shaped()
907 gtl::ArraySlice<int64_t> new_sizes) const { in shaped()
917 gtl::ArraySlice<int64_t> new_sizes) const { in bit_casted_shaped()
926 gtl::ArraySlice<int64_t> new_sizes) const { in unaligned_shaped()
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_transpose_copy_test.cpp | 47 const std::vector<int32_t> new_sizes = {3, 2}; in TEST_F() local
72 const std::vector<int32_t> new_sizes = {3, 2}; in TEST_F() local
98 const std::vector<int32_t> new_sizes = {2, 3, 1}; in TEST_F() local
130 const std::vector<int32_t> new_sizes = {3, 2, 2}; in TEST_F() local
|
| H A D | op_permute_copy_test.cpp | 61 const std::vector<int32_t> new_sizes = {4, 1, 5}; in TEST_F() local
88 const std::vector<int32_t> new_sizes = {3, 2}; in TEST_F() local
117 const std::vector<int32_t> new_sizes = {3, 2, 1}; in TEST_F() local
172 const std::vector<int32_t> new_sizes = {2, 4, 3, 3}; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Integration.cpp | 101 SymDimVector new_sizes(y.dim(), 1); // shape = [1] * y. in trapezoid() local
108 SymDimVector new_sizes = add_padding_to_shape(x.sym_sizes(), y.dim()); in trapezoid() local
147 SymDimVector new_sizes(y.dim(), 1); // shape = [1] * y. in cumulative_trapezoid() local
152 SymDimVector new_sizes = add_padding_to_shape(x.sym_sizes(), y.dim()); in cumulative_trapezoid() local
|
| H A D | NamedTensor.cpp | 237 std::vector<int64_t> new_sizes(out_dim, 1); in align_to() local
286 std::vector<int64_t> new_sizes(names.size(), 1); in align_to() local
|
| /aosp_15_r20/external/executorch/runtime/core/portable_type/test/ |
| H A D | tensor_impl_test.cpp | 77 SizesType new_sizes[RANK] = {0, 0, 0, 0, 0}; in TEST_F() local
289 SizesType new_sizes[3] = {3, 2, 4}; in TEST_F() local
318 SizesType new_sizes[3] = {2, 2, 2}; in TEST_F() local
347 SizesType new_sizes[3] = {0, 4, 4}; in TEST_F() local
446 SizesType new_sizes[2] = {3, 2}; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorTransformerFunctions.cpp | 69 Tensor new_sizes = nt_input->get_nested_sizes().clone(); in nested_linear() local
83 Tensor new_sizes = nt_self->get_nested_sizes().clone(); in NestedTensor_matmul() local
118 Tensor new_sizes = nt_mat1->get_nested_sizes().clone(); in NestedTensor_times_Tensor_plus_Tensor_addmm() local
|
| H A D | NestedTensorUtils.cpp | 95 auto new_sizes = sizes.clone(); in chunk_nested_tensor() local
152 auto new_sizes = sizes.clone(); in split_with_sizes_nested() local
|
| H A D | NestedTensorMath.cpp | 347 auto new_sizes = at::empty({sizes.size(0)}, sizes.options()); in NestedTensor_embedding() local
450 auto new_sizes = at::empty({ntensors, ndims-1}, TensorOptions().dtype(kLong)); in select_nested() local
1052 auto new_sizes = get_nested_tensor_impl(tensors[0].get())->get_nested_sizes().clone(); in cat_nested_as_jagged() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/ |
| H A D | Tensor.cpp | 400 void vTensor::update_size_metadata(const std::vector<int64_t>& new_sizes) { in update_size_metadata()
423 void vTensor::reallocate(const std::vector<int64_t>& new_sizes) { in reallocate()
431 void vTensor::virtual_resize(const std::vector<int64_t>& new_sizes) { in virtual_resize()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | transpose_util.h | 58 const SizesType* new_sizes, in increment_index_and_offset()
94 SizesType new_sizes[kTensorDimensionLimit]; in transpose_tensors() local
|
| /aosp_15_r20/external/executorch/runtime/core/exec_aten/util/ |
| H A D | tensor_util_portable.cpp | 202 exec_aten::ArrayRef<exec_aten::SizesType> new_sizes) { in resize_tensor_impl()
209 torch::executor::ArrayRef<exec_aten::SizesType> new_sizes) { in resize_tensor_impl()
|
| H A D | tensor_util.h | 1169 executorch::aten::ArrayRef<executorch::aten::SizesType> new_sizes) { in resize_tensor()
1189 executorch::aten::ArrayRef<T> new_sizes) { in resize_tensor()
1206 executorch::aten::ArrayRef<executorch::aten::SizesType> new_sizes) { in resize()
|
| H A D | tensor_util_aten.cpp | 199 c10::ArrayRef<exec_aten::SizesType> new_sizes) { in resize_tensor_impl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | TensorAccessor.h | 196 index_t* new_sizes = this->sizes_ + 1; variable
202 const index_t* new_sizes = this->sizes_ + 1; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | Linear.cpp | 140 std::vector<int64_t> new_sizes(3, 1); in apply_impl_helper() local
315 std::vector<int64_t> new_sizes(3, 1); in apply_impl() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/ |
| H A D | Tensor.cpp | 736 const std::vector<int64_t>& new_sizes, in virtual_reconfigure()
758 void vTensor::virtual_resize(const std::vector<int64_t>& new_sizes) { in virtual_resize()
|
| /aosp_15_r20/external/executorch/runtime/core/exec_aten/testing_util/test/ |
| H A D | tensor_factory_test.cpp | 1079 int32_t new_sizes[2] = {1, 1}; in TEST_F() local
1099 int32_t new_sizes[2] = {1, 1}; in TEST_F() local
1120 int32_t new_sizes[2] = {1, 1}; in TEST_F() local
|
| /aosp_15_r20/external/executorch/runtime/core/portable_type/ |
| H A D | tensor_impl.cpp | 74 Error TensorImpl::internal_resize_contiguous(ArrayRef<SizesType> new_sizes) { in internal_resize_contiguous()
|
| H A D | tensor_impl.h | 211 void set_sizes_contiguous(ArrayRef<SizesType> new_sizes) { in set_sizes_contiguous()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_diagonal_copy.cpp | 42 int64_t new_sizes[kTensorDimensionLimit]; in diagonal_copy_impl() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Transpose.cpp | 32 std::vector<int64_t> new_sizes = in->sizes(); in resize_transpose_view_node() local
|
| H A D | RepeatInterleave.cpp | 29 std::vector<int64_t> new_sizes = in->sizes(); in resize_repeat_interleave_node() local
|
| H A D | Reduce.cpp | 30 std::vector<int64_t> new_sizes = in->sizes(); in resize_reduce_node() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchedTensorImpl.cpp | 48 c10::SymDimVector new_sizes; in refreshTensorMetadata() local
|