Searched refs:lazy_cholesky_inverse_kernel (Results 1 – 1 of 1) sorted by relevance
58 Tensor& lazy_cholesky_inverse_kernel(Tensor &result, Tensor& infos, bool upper) { in lazy_cholesky_inverse_kernel() function141 REGISTER_CUDA_DISPATCH(cholesky_inverse_stub, &lazy_cholesky_inverse_kernel);