Home
last modified time | relevance | path

Searched refs:create_gradient_edge (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/autograd/functions/
Dutils.cpp33 autograd::create_gradient_edge(variable, grad_fn); in wrap_outputs()
/external/pytorch/torch/csrc/autograd/
Dfunction.h727 inline void create_gradient_edge( in create_gradient_edge() function
/external/pytorch/torch/csrc/jit/runtime/
Dgraph_executor.cpp353 autograd::create_gradient_edge(output, shared_from_this()); in addInputVariable()