Searched refs:IsPrimitiveEquals (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | auto_monad.cc | 136 return (GetSideEffectPropagate(prim) != 0) || IsPrimitiveEquals(prim, prim::kPrimRefToEmbed) || in IsKeepRef()
137 IsPrimitiveEquals(prim, prim::kPrimPull); in IsKeepRef()
571 if (IsPrimitiveEquals(prim, prim::kPrimMakeTuple)) { in TraceTupleCNodeEffectInfo()
601 if (IsPrimitiveEquals(prim, prim::kPrimTupleGetItem)) { in TraceTupleCNodeEffectInfo()
675 if (IsPrimitiveEquals(prim, prim::kPrimSwitch)) { in TraceEffectInfo()
680 if (IsPrimitiveEquals(prim, prim::kPrimSwitchLayer)) { in TraceEffectInfo()
685 if (IsPrimitiveEquals(prim, prim::kPrimTupleGetItem)) { in TraceEffectInfo()
834 if (IsPrimitiveEquals(prim, prim::kPrimReturn)) { in DetectEffectInfo()
838 if (IsPrimitiveEquals(prim, prim::kPrimCall)) { in DetectEffectInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | prim_bprop_optimizer.cc | 198 bool hookback_flg = IsPrimitiveEquals(prim, prim::kPrimHookBackward); in OptimizeBPropFuncGraph()
199 …if (hookback_flg || IsPrimitiveEquals(prim, prim::kPrimMakeTuple) || IsPrimitiveEquals(prim, prim:… in OptimizeBPropFuncGraph()
|
| D | kpynative.cc | 390 …if (IsPrimitiveEquals(prim, prim::kPrimStopGradient) || IsPrimitiveEquals(prim, prim::kPrimUpdateS… in KPynativeOp()
395 if (IsPrimitiveEquals(prim, prim::kPrimHookBackward)) { in KPynativeOp()
397 …} else if (IsPrimitiveEquals(prim, prim::kPrimMakeTuple) || IsPrimitiveEquals(prim, prim::kPrimMak… in KPynativeOp()
951 if (IsPrimitiveEquals(prim, prim::kPrimMakeTuple)) { in BuildMakeSequenceBprop()
953 } else if (IsPrimitiveEquals(prim, prim::kPrimMakeList)) { in BuildMakeSequenceBprop()
|
| D | kprim.cc | 459 if (lift_fv_before_grad && IsPrimitiveEquals(prim, prim::kPrimSwitch)) { in KPrimitive()
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | anf_model_parser.cc | 869 …if (IsPrimitiveEquals(prim::kPrimSwitch, prim) || IsPrimitiveEquals(prim::kPrimSwitchLayer, prim) … in SetCNodeAbastract()
870 IsPrimitiveEquals(prim::kPrimPartial, prim)) { in SetCNodeAbastract()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | anf.cc | 220 bool IsPrimitiveEquals(const PrimitivePtr &prim1, const PrimitivePtr &prim2) { in IsPrimitiveEquals() function
|
| D | anf.h | 583 bool IsPrimitiveEquals(const PrimitivePtr &prim1, const PrimitivePtr &prim2);
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_auto_monad.cc | 1813 if (!IsPrimitiveEquals(prim, prim::kPrimAssign)) { in IsOptimizableAssign()
|