Searched refs:kPrimHookBackward (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | dfunctor.cc | 213 if (IsPrimitiveCNode(input, prim::kPrimHookBackward)) { in BackPropagate()
265 if (IsPrimitiveCNode(node, prim::kPrimHookBackward)) { in MapMorphism()
433 if (IsPrimitiveCNode(output_node, prim::kPrimHookBackward)) { in MapMorphism()
636 …(prim->Hash() == prim::kPrimHookBackward->Hash() && prim->name() == prim::kPrimHookBackward->name(… in MapValueObject()
|
| D | kprim.cc | 394 …if (prim->Hash() == prim::kPrimHookBackward->Hash() && prim->name() == prim::kPrimHookBackward->na… in GetBprop()
|
| D | prim_bprop_optimizer.cc | 198 bool hookback_flg = IsPrimitiveEquals(prim, prim::kPrimHookBackward); in OptimizeBPropFuncGraph()
|
| D | kpynative.cc | 395 if (IsPrimitiveEquals(prim, prim::kPrimHookBackward)) { in KPynativeOp()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 65 {prim::kPrimInsertGradientOf, prim::kPrimStopGradient, prim::kPrimHookBackward, in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | special_op_eliminate.h | 46 hook_backward_(std::make_shared<PrimEliminater>(prim::kPrimHookBackward)), in SpecialOpEliminater()
66 if (IsPrimitiveCNode(node, prim::kPrimHookBackward)) { in operator()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 579 inline const PrimitivePtr kPrimHookBackward = std::make_shared<Primitive>("HookBackward"); variable
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 2469 auto fake_prim = std::make_shared<PrimitivePy>(prim::kPrimHookBackward->name()); in DoGradForCustomBprop()
|