Home
last modified time | relevance | path

Searched defs:D_grad (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/
D_lobpcg.py16 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