Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/runtime/
Dgraph_executor_impl.h33 bool needsGradient(const std::shared_ptr<const Graph>& graph);
Dgraph_executor.cpp742 if (needsGradient(opt_graph)) { in compileSpec()
906 bool needsGradient(const std::shared_ptr<const Graph>& graph) { in needsGradient() function
/external/pytorch/torch/csrc/jit/
DOVERVIEW.md1085 … actual requires a gradient to be calculated, which is determined by `needsGradient`. In the case …