Searched refs:cholesky_inverse_kernel_impl (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebraKernel.cpp | 129 Tensor& cholesky_inverse_kernel_impl(Tensor& result, Tensor& infos, bool upper) { in cholesky_inverse_kernel_impl() function
1144 REGISTER_ARCH_DISPATCH(cholesky_inverse_stub, DEFAULT, &cholesky_inverse_kernel_impl);
1145 REGISTER_AVX512_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
1146 REGISTER_AVX2_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
1147 REGISTER_VSX_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
1148 REGISTER_ZVECTOR_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | BatchLinearAlgebra.cpp | 1432 Tensor& cholesky_inverse_kernel_impl(Tensor &result, Tensor& infos, bool upper) { in cholesky_inverse_kernel_impl() function
1457 REGISTER_CUDA_DISPATCH(cholesky_inverse_stub, &cholesky_inverse_kernel_impl);
|