Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dkpynative.cc544 cnode->set_stop_gradient(true); in ForgeMakeSequenceAdjoint()
888 cnode->set_stop_gradient(true); in PropagateStopGradient()
Ddfunctor.cc734 cnode->set_stop_gradient(true); in BroadCastStopFlag()
/third_party/mindspore/mindspore/core/ir/
Danf.h285 void set_stop_gradient(bool stop_gradient) { stop_gradient_ = stop_gradient; } in set_stop_gradient() function