| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | load_eliminate.cc | 49 if (IsPrimitiveCNode(input_load, prim::kPrimLoad)) { in operator ()()
50 auto load_prim = NewValueNode(prim::kPrimLoad); in operator ()()
|
| D | updatestate_eliminate.cc | 206 if (IsPrimitiveCNode(attach, prim::kPrimLoad)) { in GetLoadsFromUpdateState()
233 …return (IsPrimitiveCNode(node, prim::kPrimLoad) && node->cast<CNodePtr>()->input(kAttachIndex) == … in GetLoadsFollowTuple()
444 if (IsPrimitiveCNode(ref_node, prim::kPrimLoad)) { in EliminateLoadBeforeAssigns()
638 if (IsPrimitiveCNode(attach, prim::kPrimPartial) || IsPrimitiveCNode(attach, prim::kPrimLoad)) { in operator ()()
774 … if (IsPrimitiveCNode(attach, prim::kPrimLoad) || IsPrimitiveCNode(attach, prim::kPrimMakeTuple)) { in operator ()()
|
| D | less_batch_normalization.cc | 251 prim::kPrimLoad, prim::kPrimRefToEmbed, prim::kPrimApplyMomentum, prim::kPrimMomentum,
280 if (IsPrimitiveCNode(cnode, prim::kPrimLoad) || IsPrimitiveCNode(cnode, prim::kPrimRefToEmbed)) { in ConvertRemoveNodeToVirtualNode()
445 … [](const AnfNodePtr &node) { return IsPrimitiveCNode(node, prim::kPrimLoad); }); in operator ()()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | order_enforce.cc | 72 … if (IsPrimitiveCNode(attach, prim::kPrimLoad) && IsPrimitiveCNode(attach, prim::kPrimMakeTuple)) { in HandleUpdateState()
84 return (user_node != update_state) && !IsPrimitiveCNode(user_node, prim::kPrimLoad); in HandleUpdateState()
95 … [](const AnfNodePtr &input) { return IsPrimitiveCNode(input, prim::kPrimLoad); }); in HasLoadInput()
126 if (IsPrimitiveCNode(input, prim::kPrimLoad)) { in FindLastUpdateState()
340 return IsPrimitiveCNode(user_node, prim::kPrimLoad); in FindLoadNodes()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | combine_cast_fusion.cc | 40 if (input0->isa<Parameter>() || (IsPrimitiveCNode(input0, prim::kPrimLoad) && in GetDealList()
46 if (IsPrimitiveCNode(input0, prim::kPrimLoad)) { in GetDealList()
56 if (IsPrimitiveCNode(input0, prim::kPrimLoad)) { in GetDealList()
|
| D | matmul_biasadd_fusion.cc | 56 VectorRef load_w = VectorRef({prim::kPrimLoad, w_, u_}); in DefinePattern()
57 VectorRef load_bias = VectorRef({prim::kPrimLoad, bias_, u_}); in DefinePattern()
|
| D | adam_fusion.cc | 91 VectorRef load_param = VectorRef({prim::kPrimLoad, param_, u_}); in DefinePattern()
92 VectorRef load_m = VectorRef({prim::kPrimLoad, m_, u_}); in DefinePattern()
93 VectorRef load_v = VectorRef({prim::kPrimLoad, v_, u_}); in DefinePattern()
|
| D | adam_weight_decay_fusion.cc | 91 VectorRef load_param = VectorRef({prim::kPrimLoad, param_, u_}); in DefinePattern()
92 VectorRef load_m = VectorRef({prim::kPrimLoad, m_, u_}); in DefinePattern()
93 VectorRef load_v = VectorRef({prim::kPrimLoad, v_, u_}); in DefinePattern()
|
| D | apply_momentum_weight_fusion.cc | 28 VectorRef load_para = VectorRef({prim::kPrimLoad, variable_, monad_}); in DefinePattern()
|
| D | apply_momentum_weight_scale_fusion.cc | 87 VectorRef load_para = VectorRef({prim::kPrimLoad, variable_, monad_}); in DefinePattern()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | anf.cc | 328 if (IsPrimitiveCNode(input, prim::kPrimLoad)) { in GetLoadInputs()
404 if (IsPrimitiveCNode(node, prim::kPrimDepend) || IsPrimitiveCNode(node, prim::kPrimLoad)) { in GetVirtualNodeTargetFromInputs()
531 IsPrimitiveCNode(node, prim::kPrimDepend) || IsPrimitiveCNode(node, prim::kPrimLoad) || in GetOriginNodeTarget()
537 if (IsPrimitiveCNode(node, prim::kPrimDepend) || IsPrimitiveCNode(node, prim::kPrimLoad) || in GetOriginNodeTarget()
|
| /third_party/mindspore/mindspore/lite/tools/converter/import/ |
| D | mindspore_importer.cc | 133 opt::CheckPrimitiveType(output_node, prim::kPrimLoad)) { in GetFuncGraphOutputName()
146 opt::CheckPrimitiveType(output_node, prim::kPrimLoad)) { in GetFuncGraphOutputName()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | auto_monad_eliminate.cc | 51 if (!IsPrimitiveCNode(cnode, prim::kPrimLoad)) { in GenerateLoadGroups()
261 if (!IsPrimitiveCNode(load_node, prim::kPrimLoad)) { in ReplaceUpdateStateForLoad()
|
| D | irpass.h | 208 return IsPrimitiveCNode(node, prim::kPrimLoad); in IsLoad()
|
| D | irpass.cc | 198 …inater_ = MakeSubstitution(std::make_shared<LoadEliminater>(), "load_eliminater", prim::kPrimLoad); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | lstm_adjust_pass.cc | 49 …iveType(weight_node, opt::kPrimIdentity) && !opt::CheckPrimitiveType(weight_node, prim::kPrimLoad); in GetRealLstmWeightNode()
60 !opt::CheckPrimitiveType(weight_node, prim::kPrimLoad); in GetRealLstmWeightNode()
|
| D | parser_utils.cc | 159 …iveType(weight_node, opt::kPrimIdentity) && !opt::CheckPrimitiveType(weight_node, prim::kPrimLoad); in GetRealConvWeightNode()
169 !opt::CheckPrimitiveType(weight_node, prim::kPrimLoad); in GetRealConvWeightNode()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline.cc | 529 if (IsPrimitiveCNode(weight_node, prim::kPrimLoad)) { in GetWeightInfo()
557 if (x->isa<Parameter>() || IsPrimitiveCNode(x, prim::kPrimLoad)) { in GetWeightInfo()
566 …if (cnode == nullptr || IsPrimitiveCNode(cnode, prim::kPrimLoad) || cnode->size() != expect_input_… in GetWeightInfo()
571 …(!fakequant_min_node->isa<Parameter>() && !IsPrimitiveCNode(fakequant_min_node, prim::kPrimLoad)) { in GetWeightInfo()
575 if (IsPrimitiveCNode(fakequant_min_node, prim::kPrimLoad)) { in GetWeightInfo()
631 if (!weight_node->isa<Parameter>() && !IsPrimitiveCNode(weight_node, prim::kPrimLoad)) { in FetchInfoForQuantExport()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_n_optimizer.cc | 113 if (in_node_name == prim::kPrimDepend->name() || in_node_name == prim::kPrimLoad->name()) { in InputCheck()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | optimize_assign.cc | 105 AnfNodePtrList load_inputs = {NewValueNode(prim::kPrimLoad), assign_to_node, update_state_node}; in KeepExecOrder()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | optimize_dependence.cc | 55 AnfAlgo::GetCNodeName(cnode) != prim::kPrimLoad->name()) { in CheckIsolatedVirtualNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 94 while (IsPrimitiveCNode(temp, prim::kPrimLoad) || IsPrimitiveCNode(temp, prim::kPrimCast)) { in NeedGrad()
378 if (IsPrimitiveCNode(node, prim::kPrimLoad) || IsPrimitiveCNode(node, prim::kPrimCast)) { in GetActualOpUsers()
721 IsPrimitiveCNode(node, prim::kPrimLoad)) { in ActualOp()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | convert.cc | 1051 if ((name == prim::kPrimLoad->name()) || (name == prim::kPrimDepend->name()) || in IsControlEdgeNode()
1150 if (name == prim::kPrimLoad->name()) { in AutoMonadCollectInput()
1267 if (IsPrimitiveCNode(pred, prim::kPrimLoad)) { in GetRealInputNode()
1424 if (IsValueNode<Primitive>(node) || IsPrimitiveCNode(node, prim::kPrimLoad) || in Convert()
1453 if (IsPrimitiveCNode(item, prim::kPrimLoad)) { in ConvertMakeTuple()
|
| /third_party/mindspore/tests/ut/cpp/ir/ |
| D | manager_test.cc | 635 auto x1 = fg->NewCNode({NewValueNode(prim::kPrimLoad), x, u}); in TEST_F()
693 auto x1 = fg->NewCNode({NewValueNode(prim::kPrimLoad), x, u}); in TEST_F()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 171 {prim::kPrimLoad, {InferImplLoad, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|