| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | ExpandUtils.cpp | 63 IntArrayRef tensor_sizes, in inferExpandGeometryImpl()
117 IntArrayRef tensor_sizes, in inferExpandGeometry()
126 IntArrayRef tensor_sizes, in inferExpandGeometry_dimvector()
148 std::vector<int64_t> infer_dense_strides(IntArrayRef tensor_sizes, IntArrayRef tensor_strides) { in infer_dense_strides()
|
| H A D | WrapDimUtils.h | 32 const std::vector<std::vector<int64_t>>& tensor_sizes) { in maybe_wrap_dim()
109 const std::vector<std::vector<int64_t>>& tensor_sizes) { in legacy_cat_wrap_dim()
121 const std::vector<std::vector<c10::SymInt>>& tensor_sizes) { in legacy_cat_wrap_dim_symint()
|
| H A D | LegacyVmapTransforms.cpp | 85 auto tensor_sizes = tensor_.sizes(); in getPhysicalShape() local
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | nn_utils.cpp | 615 std::vector<int64_t> tensor_sizes{seq_len, 5}; in TEST_F() local
654 std::vector<int64_t> tensor_sizes{length - tensor.size(0)}; in TEST_F() local
803 std::vector<int64_t> tensor_sizes{length - tensor.size(0)}; in TEST_F() local
853 std::vector<int64_t> tensor_sizes{seq_len, 5}; in TEST_F() local
|
| H A D | tensor_indexing.cpp | 553 std::vector<int64_t> tensor_sizes{5, 1}; in TEST() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | NamedTensor.cpp | 82 IntArrayRef tensor_sizes, in aligned_size()
194 const auto tensor_sizes = tensor.sizes(); in align_to() local
284 auto tensor_sizes = tensor.sizes(); in align_to() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | LegacyVmapTransforms.cpp | 72 auto tensor_sizes = tensor_.sizes(); in getPhysicalShape() local
81 auto tensor_sizes = tensor_.sym_sizes(); in getPhysicalShape() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorUtils.h | 184 std::vector<IntArrayRef> tensor_sizes = in get_size_for_index() local
411 auto tensor_sizes = tensor_node.children(i).sizes(); in wrap_tensor_node() local
|
| H A D | NestedTensorMath.cpp | 301 IntArrayRef tensor_sizes(sizes_ptr, sizes_num_columns); in NestedTensor_to_padded_tensor_generic() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/ |
| H A D | tfrt_fallback.cc | 41 auto tensor_sizes = tensor.shape().dim_sizes(); in ConvertTensorToPyArray() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Blas.cpp | 64 IntArrayRef tensor_sizes = tensor.sizes(); in prepare_matrix_for_cublas() local
83 IntArrayRef tensor_sizes = tensor.sizes(); in prepare_matrix_for_cublas() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/ |
| H A D | executor.cpp | 44 auto tensor_sizes = arg.sizes().vec(); in getMapSize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | pooling_test.cc | 52 std::vector<int64_t> tensor_sizes(num_spatial_dims + 2, 1); in ExpandWithBatchAndFeatureDimensions() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorTransformerUtils.cpp | 88 const Tensor& tensor_sizes = tensor->get_nested_sizes(); in is_safe_to_get_storage_as_tensor() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | graph_fuser.cpp | 849 auto tensor_sizes = fmap(tensor_inputs, [&](Value* v) { in tryToMoveChunk() local
|