Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/
Devaluator.cc195 auto stop_gradient_node = cnode->input(stop_gradient_pos); in HasIsolatedSideEffectNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.cc972 … auto stop_gradient_node = fg->NewCNodeInOrder({NewValueNode(prim::kPrimStopGradient), aux_node}); in GenerateFuncGraph() local
980 … auto stop_gradient_node = fg->NewCNodeInOrder({NewValueNode(prim::kPrimStopGradient), aux_node}); in GenerateFuncGraph() local