Searched refs:kPrimStopGradient (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | stopgrad_eliminate.h | 47 if (IsPrimitiveCNode(input, prim::kPrimStopGradient)) { in GetInputStopGradient()
|
| D | special_op_eliminate.h | 45 stop_gradient_(std::make_shared<PrimEliminater>(prim::kPrimStopGradient)), in SpecialOpEliminater()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | pynative_dfunctor.cc | 132 if (IsPrimitiveCNode(cnode_morph, prim::kPrimStopGradient)) { in RunOutputReplace()
183 IsPrimitiveCNode(input_node, prim::kPrimStopGradient)) { in RunInputReplace()
|
| D | dfunctor.cc | 524 …if ((prim->Hash() == prim::kPrimStopGradient->Hash() && prim->name() == prim::kPrimStopGradient->n… in MapPrimitiveToK()
731 …if (IsPrimitiveCNode(cnode, prim::kPrimStopGradient) || IsPrimitiveCNode(cnode, prim::kPrimUpdateS… in BroadCastStopFlag()
|
| D | kpynative.cc | 390 …if (IsPrimitiveEquals(prim, prim::kPrimStopGradient) || IsPrimitiveEquals(prim, prim::kPrimUpdateS… in KPynativeOp()
885 …if (IsPrimitiveCNode(cnode, prim::kPrimStopGradient) || IsPrimitiveCNode(cnode, prim::kPrimUpdateS… in PropagateStopGradient()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 65 {prim::kPrimInsertGradientOf, prim::kPrimStopGradient, prim::kPrimHookBackward, in OptimizeIRPassLib()
202 …tution(std::make_shared<StopGradientEliminater>(), "stopgrad_eliminater", prim::kPrimStopGradient); in OptimizeIRPassLib()
|
| /third_party/mindspore/tests/ut/cpp/optimizer/ad/ |
| D | kpynative_test.cc | 72 auto b_node = g->NewCNode({NewValueNode(prim::kPrimStopGradient), a_node}); in BuildStopGradient()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | auto_monad.cc | 1197 IsPrimitiveCNode(output->cast<CNodePtr>()->input(2), prim::kPrimStopGradient)) { in ClearIsolatedNodes()
1379 …PrimitiveCNode(output->cast<CNodePtr>()->input(kDependAttachNodeIndex), prim::kPrimStopGradient)) { in AttachToOutput()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/ |
| D | function_block.cc | 544 AnfNodePtr stop_grad_node = func_graph_->NewCNode({NewValueNode(prim::kPrimStopGradient), state}); in AttachIsolatedNodesBeforeReturn()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/ |
| D | ops_front_infer_function.cc | 668 REGISTER_PRIMITIVE_FRONT_EVAL_IMPL(StopGradient, prim::kPrimStopGradient, InferImplStopGradient, nu…
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 710 inline const PrimitivePtr kPrimStopGradient = std::make_shared<Primitive>("stop_gradient"); variable
|