Home
last modified time | relevance | path

Searched defs:LU_strides (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebra.cpp253 auto LU_strides = at::native::batched_matrix_contiguous_strides(shape, /*f-contig*=*/true); in TORCH_META_FUNC() local
276 auto LU_strides = at::native::batched_matrix_contiguous_strides(shape, /*f-contig*=*/true); in TORCH_META_FUNC() local
H A DBatchLinearAlgebra.cpp536 auto LU_strides = at::native::batched_matrix_contiguous_strides(shape, /*f-contig*=*/true); in TORCH_META_FUNC() local
566 auto LU_strides = at::native::batched_matrix_contiguous_strides(sizes, /*f-contig*=*/true); in TORCH_META_FUNC() local