Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebraUtils.h436 inline std::vector<int64_t> create_dim_backshift_permutation(int64_t dim0, int64_t dim1, int64_t nd… in create_dim_backshift_permutation() function
H A DLinearAlgebra.cpp2906 auto permutation = create_dim_backshift_permutation(dim_[0], dim_[1], A.dim()); in linalg_matrix_norm()
2967 auto permutation = create_dim_backshift_permutation(dim_[0], dim_[1], A_.dim()); in linalg_matrix_norm()