Home
last modified time | relevance | path

Searched refs:kPrimDepend (Results 1 – 25 of 65) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dadam_apply_one_with_decay_rule.cc39 if (AnfAlgo::CheckPrimitiveType(final_node, prim::kPrimDepend)) { in GetFusionNodeInputs()
165 VectorRef depend0 = VectorRef({prim::kPrimDepend, sub0, assign0}); in DefinePattern()
167 VectorRef depend1 = VectorRef({prim::kPrimDepend, depend0, assign1}); in DefinePattern()
169 return VectorRef({prim::kPrimDepend, depend1, assign2}); in DefinePattern()
190 VectorRef depend0 = VectorRef({prim::kPrimDepend, sub0, assign0}); in DefinePattern()
192 VectorRef depend1 = VectorRef({prim::kPrimDepend, depend0, assign1}); in DefinePattern()
194 return VectorRef({prim::kPrimDepend, depend1, assign2}); in DefinePattern()
215 VectorRef depend0 = VectorRef({prim::kPrimDepend, sub0, assign0}); in DefinePattern()
217 VectorRef depend1 = VectorRef({prim::kPrimDepend, depend0, assign1}); in DefinePattern()
219 return VectorRef({prim::kPrimDepend, depend1, assign2}); in DefinePattern()
[all …]
Dadam_apply_one_fusion.cc112 VectorRef depend0 = VectorRef({prim::kPrimDepend, sub0, assign0}); in DefinePattern()
114 VectorRef depend1 = VectorRef({prim::kPrimDepend, depend0, assign1}); in DefinePattern()
116 return VectorRef({prim::kPrimDepend, depend1, assign2}); in DefinePattern()
134 VectorRef depend0 = VectorRef({prim::kPrimDepend, sub0, assign0}); in DefinePattern()
136 VectorRef depend1 = VectorRef({prim::kPrimDepend, depend0, assign1}); in DefinePattern()
138 return VectorRef({prim::kPrimDepend, depend1, assign2}); in DefinePattern()
156 VectorRef depend0 = VectorRef({prim::kPrimDepend, sub0, assign0}); in DefinePattern()
158 VectorRef depend1 = VectorRef({prim::kPrimDepend, depend0, assign1}); in DefinePattern()
160 return VectorRef({prim::kPrimDepend, depend1, assign2}); in DefinePattern()
178 VectorRef depend0 = VectorRef({prim::kPrimDepend, sub0, assign0}); in DefinePattern()
[all …]
Dfused_batch_norm_fusion.cc259 VectorRef depend0 = VectorRef({prim::kPrimDepend, tuple_getitem0, assign_sub0}); in DefinePattern()
260 return VectorRef({prim::kPrimDepend, depend0, assign_sub1}); in DefinePattern()
282 VectorRef depend0 = VectorRef({prim::kPrimDepend, tuple_getitem0, assign_sub0}); in DefinePattern()
283 return VectorRef({prim::kPrimDepend, depend0, assign_sub1}); in DefinePattern()
305 VectorRef depend0 = VectorRef({prim::kPrimDepend, tuple_getitem0, assign_sub0}); in DefinePattern()
306 return VectorRef({prim::kPrimDepend, depend0, assign_sub1}); in DefinePattern()
Dmomentum_lossscale_fusion.cc78 auto depend_prim = NewValueNode(prim::kPrimDepend); in Process()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dcast_eliminate.cc30 while (IsPrimitiveCNode(cur_node, prim::kPrimDepend)) { in TransThroughDepend()
66 if (IsPrimitiveCNode(node->cast<CNodePtr>()->input(2), prim::kPrimDepend)) { in operator ()()
68 …node->func_graph()->NewCNode({NewValueNode(prim::kPrimDepend), src_, node->cast<CNodePtr>()->input… in operator ()()
Dbranch_culling.cc105 std::vector<PrimitivePtr> adapter_convert_ops = {prim::kPrimDepend, prim::kPrimLoad}; in InConvertWhiteList()
281 AnfNodePtrList inputs = {NewValueNode(prim::kPrimDepend), square_op, output_node}; in GenerateSwitchDependNode()
313 if (IsPrimitiveCNode(item.first, prim::kPrimDepend)) { in IsNetOutputNode()
336 if (IsPrimitiveCNode(depended_tuple_input_node->cast<CNodePtr>(), prim::kPrimDepend)) { in GenerateReplNodeForDependMakeTuple()
404 if (IsPrimitiveCNode(node, prim::kPrimDepend)) { in TransformGraphDependNode()
414 if (IsPrimitiveCNode(depended_node, prim::kPrimDepend)) { in TransformGraphDependNode()
Dbranch_culling.h139 …MATCH_REPLACE_IF(node, PPrimitive(prim::kPrimSwitch, PPrimitive(prim::kPrimDepend, v, x), true_br,… in operator()
140 … PPrimitive(prim::kPrimDepend, PPrimitive(prim::kPrimSwitch, v, true_br, false_br), x), in operator()
Drecompute_prepare.h63 …if (IsPrimitiveCNode(output_cnode, prim::kPrimDepend) || IsPrimitiveCNode(output_cnode, prim::kPri… in GetRealOutputNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dadjust_depend_for_parallel_optimizer_recompute_all_gather.cc106 if (IsPrimitiveCNode(depend_node, prim::kPrimDepend)) { in AdjustAllgatherDepend()
115 …:vector<AnfNodePtr> inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend->name())), in AdjustAllgatherDepend()
124 … IsPrimitiveCNode(AnfAlgo::GetInputNode(depend_node->cast<CNodePtr>(), 0), prim::kPrimDepend)) { in AdjustAllgatherDepend()
135 …:vector<AnfNodePtr> inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend->name())), in AdjustAllgatherDepend()
Deliminate_redundant_op.cc59 } else if (IsPrimitive(input0, prim::kPrimDepend)) { in GetRealPrevCNode()
96 if (AnfAlgo::CheckPrimitiveType(nd, prim::kPrimDepend)) { in ProcessMatchedNodes()
Doptimize_dependence.cc54 if (AnfAlgo::GetCNodeName(cnode) != prim::kPrimDepend->name() && in CheckIsolatedVirtualNode()
161 if (!cnode->IsApply(prim::kPrimDepend) && !cnode->IsApply(prim::kPrimUpdateState)) { in SearchTransDataAndCast()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadam_fusion.cc111 next_param = VectorRef({prim::kPrimDepend, next_param, assign_param}); in DefinePattern()
115 next_param = VectorRef({prim::kPrimDepend, next_param, assign_m}); in DefinePattern()
118 next_param = VectorRef({prim::kPrimDepend, next_param, assign_v}); in DefinePattern()
150 auto prim_depend = std::make_shared<Primitive>(prim::kPrimDepend->name()); in Process()
Dadam_weight_decay_fusion.cc113 next_param = VectorRef({prim::kPrimDepend, next_param, assign_param}); in DefinePattern()
117 next_param = VectorRef({prim::kPrimDepend, next_param, assign_m}); in DefinePattern()
120 next_param = VectorRef({prim::kPrimDepend, next_param, assign_v}); in DefinePattern()
155 auto prim_depend = std::make_shared<Primitive>(prim::kPrimDepend->name()); in Process()
Dapply_momentum_weight_scale_fusion.cc65 (IsPrimitiveCNode(in, prim::kPrimDepend) && in IsCast()
77 if (IsPrimitiveCNode(node, prim::kPrimDepend)) { in GetCastInput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dsplit_n_optimizer.cc113 if (in_node_name == prim::kPrimDepend->name() || in_node_name == prim::kPrimLoad->name()) { in InputCheck()
136 if (IsPrimitiveCNode(item, prim::kPrimDepend)) { in OutputCheck()
Dinsert_depend_for_all_gather.cc46 …std::vector<AnfNodePtr> inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimDepend->name… in Run()
/third_party/mindspore/mindspore/core/utils/
Dparallel_node_check.cc47 static const std::set<PrimitivePtr> TRIVIAL_NODE_LIST_ = {prim::kPrimCast, prim::kPrimDepend};
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_build.cc39 … kernel_name == prim::kPrimDepend->name() || kernel_name == prim::kPrimStateSetItem->name()) { in CreateGPUKernel()
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindspore_importer.cc106 if (!opt::CheckPrimitiveType(cnode, prim::kPrimDepend)) { in ProcessDependCnode()
154 } else if (opt::CheckPrimitiveType(output_node, prim::kPrimDepend)) { in GetFuncGraphOutputName()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dsplit_umonad.cc50 …AnfNodePtrList depend_inputs = {NewValueNode(prim::kPrimDepend), original_inputs[input_idx], origi… in ProcessNode()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Drecompute.cc229 if (IsPrimitiveCNode(cnode, prim::kPrimDepend) && i == kDependAttachNodeIndex) { in HasGradInputs()
271 …(IsPrimitiveCNode(node, prim::kPrimDepend) && WithRecomputedScope(node->input(kRealInputIndexInDep… in SetRecomputedScope()
395 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), first_input, in NewRecomputedNode()
Dauto_monad_eliminate.cc56 …} else if (IsPrimitiveCNode(input, prim::kPrimDepend) && input->cast<CNodePtr>()->input(1)->isa<Pa… in GenerateLoadGroups()
332 if (!IsPrimitiveCNode(output, prim::kPrimDepend)) { in EliminateAutoMonadNode()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc36 …prim::kPrimDepend, prim::kPrimTupleGetItem, prim::kPrimAdd, prim::kPrimSoftmaxCrossEntropyWi…
119 if (IsPrimitiveCNode(cnode, prim::kPrimDepend) && enable_parallel_optimizer) { in InsertVirtualAssignAdd()
282 …std::vector<AnfNodePtr> depend_input = {NewValueNode(prim::kPrimDepend), post_cnode->input(1), pri… in InsertDepend()
573 if (IsPrimitiveCNode(node, prim::kPrimDepend)) { in GetActualOp()
590 if (IsPrimitiveCNode(node, prim::kPrimDepend) || IsPrimitiveCNode(node, prim::kPrimZerosLike)) { in GetBorderNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc140 std::vector<AnfNodePtr> replace_depend_inputs{NewValueNode(prim::kPrimDepend), transed_node, in EliminateDependTransop()
408 if ((cnode_name == prim::kPrimDepend->name() && output_info.second == kDependAttachNodeIndex) || in GetRealNodeUsedList()
431 if ((cnode_name == prim::kPrimDepend->name() && output_info.second == kDependAttachNodeIndex) || in GetRealNodeUsedListByOutputIdx()
473 if (name == prim::kPrimDepend->name() || name == prim::kPrimMakeTuple->name() || in IsNotRealUsedByOthers()
562 if (AnfAlgo::CheckPrimitiveType(input_node, prim::kPrimDepend)) { in ConstInputToAttr()
929 if (AnfAlgo::CheckPrimitiveType(output, prim::kPrimDepend) || in TransferDependOrUpdateState()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc280 if (IsPrimitiveCNode(node, prim::kPrimDepend)) { in GraphOutNode()
524 while (IsPrimitiveCNode(output, prim::kPrimDepend)) { in FindPipelineCareNode()
542 if (IsPrimitiveCNode(cnode, prim::kPrimDepend)) { in FindPipelineCareNode()
704 if (IsPrimitiveCNode(cnode, prim::kPrimDepend)) { in Reuse()
720 if (IsPrimitiveCNode(node, prim::kPrimDepend) || IsPrimitiveCNode(node, prim::kPrimCast) || in ActualOp()
912 std::vector<AnfNodePtr> out = {NewValueNode(prim::kPrimDepend)}; in CutGraph()

123