Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorMath.cpp704 inline std::tuple<bool, Tensor, Tensor> NestedTensor_compute_size_stride( in NestedTensor_compute_size_stride() function
839 auto [viewable, sizemat_reshaped, stridemat_reshaped] = NestedTensor_compute_size_stride( in view_nested()
930 auto [viewable, sizemat_reshaped, stridemat_reshaped] = NestedTensor_compute_size_stride( in reshape_nested()