Searched defs:LU_ (Results 1 – 4 of 4) sorted by relevance
272 auto LU_ = moveBatchDimToFront(LU, LU_bdim); in linalg_lu_unpack_batch_rule() local303 auto LU_ = moveBatchDimToFront(LU, LU_bdim); in linalg_lu_solve_batch_rule() local
2435 auto LU_ = maybe_expand_lu(B, LU); in lu_solve_kernel() local2476 auto LU_ = maybe_expand_lu(B, LU); in lu_solve_kernel() local2483 auto LU_ = maybe_expand_lu(B, LU); in lu_solve_kernel() local
4195 auto LU_ = in linalg_det_jvp() local4227 auto LU_ = in linalg_det_backward() local
2128 auto LU_ = at::native::borrow_else_clone( in TORCH_IMPL_FUNC() local