Home
last modified time | relevance | path

Searched defs:tensor_sizes (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DExpandUtils.cpp63 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 DWrapDimUtils.h32 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 DLegacyVmapTransforms.cpp85 auto tensor_sizes = tensor_.sizes(); in getPhysicalShape() local
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dnn_utils.cpp615 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 Dtensor_indexing.cpp553 std::vector<int64_t> tensor_sizes{5, 1}; in TEST() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNamedTensor.cpp82 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 DLegacyVmapTransforms.cpp72 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 DNestedTensorUtils.h184 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 DNestedTensorMath.cpp301 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 Dtfrt_fallback.cc41 auto tensor_sizes = tensor.shape().dim_sizes(); in ConvertTensorToPyArray() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DBlas.cpp64 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 Dexecutor.cpp44 auto tensor_sizes = arg.sizes().vec(); in getMapSize() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dpooling_test.cc52 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 DNestedTensorTransformerUtils.cpp88 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 Dgraph_fuser.cpp849 auto tensor_sizes = fmap(tensor_inputs, [&](Value* v) { in tryToMoveChunk() local