Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dgraph_executor.cpp416 struct DifferentiableGraphOp { struct
417 DifferentiableGraphOp(Gradient grad) in DifferentiableGraphOp() function
426 void operator()(Stack& stack) const { in operator ()()
475 at::Tensor detach(at::Tensor t) const { in detach()
482 void detach(IValue& v) const { in detach()
494 void detachVariables(Stack& stack) const { in detachVariables()
505 void captureInputs( in captureInputs()
512 void captureOutputs( in captureOutputs()
520 std::shared_ptr<GraphExecutor> f_ptr;
521 Code legacy_f;
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/jit/
H A DOVERVIEW.md1304 ## DifferentiableGraphOp ## section in Executing Programs