Searched defs:U_grad (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/ |
| D | _lobpcg.py | 16 def _symeig_backward_complete_eigenspace(D_grad, U_grad, A, D, U): argument
141 def _symeig_backward_partial_eigenspace(D_grad, U_grad, A, D, U, largest): argument
248 def _symeig_backward(D_grad, U_grad, A, D, U, largest): argument
304 def backward(ctx, D_grad, U_grad): argument
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 5934 const Tensor& U_grad, in lu_unpack_backward()
6454 const Tensor& U_grad, in linalg_lu_backward()
6591 const auto U_grad = grad.narrow_symint(-2, 0, k); in lu_factor_ex_backward() local
|