Home
last modified time | relevance | path

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

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