Home
last modified time | relevance | path

Searched refs:NestedTensor_get_sizes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorUtils.h108 inline std::vector<IntArrayRef> NestedTensor_get_sizes( in NestedTensor_get_sizes() function
136 inline std::vector<IntArrayRef> NestedTensor_get_sizes(const at::Tensor& self) { in NestedTensor_get_sizes() function
138 return NestedTensor_get_sizes(self_ptr); in NestedTensor_get_sizes()
185 NestedTensor_get_sizes(get_nested_tensor_impl(tensor)); in get_size_for_index()
H A DNestedTensorMatmul.cpp82 std::vector<IntArrayRef> self_sizes = NestedTensor_get_sizes(self_ptr); in matmul_with_bmm_nested()
89 std::vector<IntArrayRef> mat2_sizes = NestedTensor_get_sizes(mat2_ptr); in matmul_with_bmm_nested()
H A DNestedTensorMath.cpp429 std::vector<IntArrayRef> sizes = NestedTensor_get_sizes(self_ptr), in select_nested()
834 std::vector<IntArrayRef> sizes = NestedTensor_get_sizes(self_ptr), in view_nested()
925 std::vector<IntArrayRef> sizes = NestedTensor_get_sizes(self_ptr), in reshape_nested()
H A DNestedTensorFactories.cpp186 std::vector<IntArrayRef> sizes = NestedTensor_get_sizes(self_ptr), in NestedTensor_unbind()