Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorUtils.h141 inline std::vector<IntArrayRef> NestedTensor_get_strides( in NestedTensor_get_strides() function
162 inline std::vector<IntArrayRef> NestedTensor_get_strides( in NestedTensor_get_strides() function
165 return NestedTensor_get_strides(self_ptr); in NestedTensor_get_strides()
195 NestedTensor_get_strides(get_nested_tensor_impl(tensor)); in get_stride_for_index()
H A DNestedTensorMatmul.cpp83 std::vector<IntArrayRef> self_strides = NestedTensor_get_strides(self_ptr); in matmul_with_bmm_nested()
90 std::vector<IntArrayRef> mat2_strides = NestedTensor_get_strides(mat2_ptr); in matmul_with_bmm_nested()
H A DNestedTensorMath.cpp430 strides = NestedTensor_get_strides(self_ptr); in select_nested()
835 strides = NestedTensor_get_strides(self_ptr); in view_nested()
926 strides = NestedTensor_get_strides(self_ptr); in reshape_nested()
H A DNestedTensorFactories.cpp187 strides = NestedTensor_get_strides(self_ptr); in NestedTensor_unbind()