Searched refs:triangular_solve_kernel (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebraKernel.cpp | 747 void triangular_solve_kernel(const Tensor& A, const Tensor& B, bool left, bool upper, TransposeType… in triangular_solve_kernel() function
1186 REGISTER_ARCH_DISPATCH(triangular_solve_stub, DEFAULT, &triangular_solve_kernel);
1187 REGISTER_AVX512_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
1188 REGISTER_AVX2_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
1189 REGISTER_VSX_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
1190 REGISTER_ZVECTOR_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebra.cpp | 1749 void triangular_solve_kernel(const Tensor& A, const Tensor& B, bool left, bool upper, TransposeType… in triangular_solve_kernel() function
1767 REGISTER_CUDA_DISPATCH(triangular_solve_stub, &triangular_solve_kernel);
2667 triangular_solve_kernel( in linalg_lstsq_gels()
2691 triangular_solve_kernel( in linalg_lstsq_gels()
|