Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/autograd/
Dautograd.cpp166 const variable_list& grad_tensors, in backward()
Dpython_engine.cpp190 PyObject* grad_tensors = nullptr; in THPEngine_run_backward() local
/external/pytorch/benchmarks/tensorexpr/
Dpt_engine.py45 def backward(self, tensors, grad_tensors, _): argument
/external/pytorch/torch/csrc/jit/runtime/
Dregister_prim_ops_fulljit.cpp272 auto grad_tensors = pop(stack); in __anon992a14c81a02() local