| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | rewrite_output_shape.cc | 25 if (!IsPrimitiveCNode(func_graph->output(), prim::kPrimMakeTuple)) { in Run()
31 if (IsPrimitiveCNode(output, prim::kPrimMakeTuple)) { in Run()
37 AnfNodePtrList mt_inputs = {NewValueNode(prim::kPrimMakeTuple), output}; in Run()
51 if (IsPrimitiveCNode(node, prim::kPrimMakeTuple)) return; in Process()
79 if (!IsPrimitiveCNode(func_graph->output(), prim::kPrimMakeTuple)) { in Run()
85 if (!IsPrimitiveCNode(output, prim::kPrimMakeTuple)) { in Run()
|
| D | eliminate_redundant_output.cc | 63 if (!IsPrimitiveCNode(output, prim::kPrimMakeTuple)) { in GetGraphKernelGetitemList()
95 IsPrimitiveCNode(AnfAlgo::GetCNodeFuncGraphPtr(node)->output(), prim::kPrimMakeTuple); in FindGraphKernelsWithMultiOutput()
230 AnfNodePtrList new_maketuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in ReplaceMakeTuple()
278 … if (!IsPrimitiveCNode(AnfAlgo::GetCNodeFuncGraphPtr(new_node)->output(), prim::kPrimMakeTuple)) { in Run()
|
| D | update_state_formatter.cc | 41 if (IsPrimitiveCNode(nodes[i], prim::kPrimMakeTuple)) { in SpreadTuples()
118 mt_inputs.insert(mt_inputs.begin(), NewValueNode(prim::kPrimMakeTuple)); in Run()
239 AnfNodePtrList mt_input = {NewValueNode(prim::kPrimMakeTuple)}; in ProcessIndex()
|
| D | decrease_compute_precision.cc | 63 if (!AnfAlgo::CheckPrimitiveType(cnode, prim::kPrimMakeTuple)) { in UpdateOutputInfo()
199 if (AnfAlgo::CheckPrimitiveType(old_output, prim::kPrimMakeTuple)) { in Process()
200 (void)new_inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/acl_vm/ |
| D | acl_multi_graph_session.cc | 104 …out, 0, false, std::vector<PrimitivePtr>{prim::kPrimMakeTuple, prim::kPrimUpdateState, prim::kPrim… in ConstructOutputRef()
106 if (AnfAlgo::CheckPrimitiveType(anf_node, prim::kPrimMakeTuple)) { in ConstructOutputRef()
135 … std::vector<PrimitivePtr>{prim::kPrimMakeTuple, prim::kPrimUpdateState, prim::kPrimStateSetItem}); in ConstructOutputRefByTupleNode()
137 if (AnfAlgo::CheckPrimitiveType(anf_node, prim::kPrimMakeTuple)) { in ConstructOutputRefByTupleNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | merge_addn.h | 53 (void)args_.insert(args_.begin(), NewValueNode(prim::kPrimMakeTuple)); in operator()
61 if (!IsPrimitiveCNode(cnode, prim::kPrimMakeTuple)) { in Visit()
180 if (!IsPrimitiveCNode(cnode, prim::kPrimMakeTuple)) { in Visit()
188 filtered_Xs_.push_back(NewValueNode(prim::kPrimMakeTuple)); in Visit()
|
| D | convert.h | 41 if (cnode->size() == 2 && IsPrimitiveCNode(cnode->input(1), prim::kPrimMakeTuple)) { in operator()
46 args.push_back(NewValueNode(prim::kPrimMakeTuple)); in operator()
|
| D | updatestate_eliminate.cc | 208 } else if (IsPrimitiveCNode(attach, prim::kPrimMakeTuple)) { in GetLoadsFromUpdateState()
352 make_tuple_inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in EliminateUpdateStateForLoads()
417 AnfNodePtrList make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), a1, assign}; in EliminateUpdateStateBetweenAssigns()
480 …if (IsPrimitiveCNode(make_tuple, prim::kPrimMakeTuple) && OnlyUsedByOneNode(make_tuple, u3_cnode))… in EliminateUpdateStateBetweenAssignMakeTuple()
510 …AnfNodePtrList new_make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), make_tuple_cnode->input(k… in EliminateUpdateStateBetweenAssignMakeTuple()
583 AnfNodePtrList new_make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), a1, in EliminateUpdateStateBetweenMakeTupleAssign()
602 if (IsPrimitiveCNode(attach, prim::kPrimMakeTuple)) { in EliminateUpdateStateForAssign()
618 if (IsPrimitiveCNode(u_attach, prim::kPrimMakeTuple)) { in EliminateUpdateStateForAssign()
650 if (IsPrimitiveCNode(attach, prim::kPrimMakeTuple)) { in operator ()()
677 … IsPrimitiveCNode(cnode, prim::kPrimPartial) || IsPrimitiveCNode(cnode, prim::kPrimMakeTuple)) { in operator ()()
[all …]
|
| D | accumulaten_eliminate.h | 65 if (!IsPrimitiveCNode(cnode, prim::kPrimMakeTuple)) { in Visit()
73 inputs_.push_back(NewValueNode(prim::kPrimMakeTuple)); in Visit()
|
| D | incorporate_getitem.h | 56 if (IsPrimitiveCNode(output, prim::kPrimMakeTuple)) { in operator()
108 if (IsPrimitiveCNode(output, prim::kPrimMakeTuple)) { in operator()
146 if (IsPrimitiveCNode(output, prim::kPrimMakeTuple)) { in IsOutputShrinkable()
156 if (IsPrimitiveCNode(output, prim::kPrimMakeTuple)) { in GetOutputSize()
224 if (IsPrimitiveCNode(new_fg_output, prim::kPrimMakeTuple)) { in ShrinkUnsedOutput()
233 AnfNodePtrList shrunk_inputs{NewValueNode({prim::kPrimMakeTuple})}; in ShrinkUnsedOutput()
308 if (IsPrimitiveCNode(output, prim::kPrimMakeTuple)) { in AlwaysTransformThisIndex()
848 if (!IsPrimitiveCNode(output, prim::kPrimMakeTuple)) { in ExistEnvNodeInTupleItem()
907 auto layers_node = fg->NewCNode(prim::kPrimMakeTuple, layers); in operator()
931 if (IsPrimitiveCNode(cnode, prim::kPrimMakeTuple) && IsValueNode<FuncGraph>(inputs[1])) { in Visit()
[all …]
|
| D | arithmetic_simplify.cc | 68 PPrimitive(prim::kPrimMakeTuple, z, y)); in operator ()()
106 auto admktup_pat = PBinOperation(prim::kPrimMakeTuple, mul_pat, z, true); in operator ()()
185 if (node != addn_maketuple && IsPrimitiveCNode(node, prim::kPrimMakeTuple)) { in ProcessDependEdge()
|
| D | branch_culling.cc | 257 …std::vector<AnfNodePtr> make_tuple_nodes{NewValueNode(prim::kPrimMakeTuple), square_op, opposite_c… in MergeNode()
333 new_make_tuple_nodes.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateReplNodeForDependMakeTuple()
374 if (IsPrimitiveCNode(depended_node, prim::kPrimMakeTuple)) { in GenerateRepDepend()
485 …std::vector<AnfNodePtr> make_tuple_nodes{NewValueNode(prim::kPrimMakeTuple), block_nodes[1], block… in GenerateMergeNodes()
496 make_tuple_nodes.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateMergeNodes()
|
| D | recompute_prepare.h | 65 } else if (IsPrimitiveCNode(output_cnode, prim::kPrimMakeTuple)) { in GetRealOutputNodes()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | kprim.cc | 344 …if (prim->Hash() == prim::kPrimMakeTuple->Hash() && prim->name() == prim::kPrimMakeTuple->name()) { in KMetaFuncGraph()
346 bprop_registry_meta_[prim::kPrimMakeTuple] = meta; in KMetaFuncGraph()
369 auto make_tuple = NewValueNode(prim::kPrimMakeTuple); in AppendMonadOutput()
431 …} else if (prim->Hash() == prim::kPrimMakeTuple->Hash() && prim->name() == prim::kPrimMakeTuple->n… in KPrimitive()
499 if (IsPrimitiveCNode(bprop_fg->output(), prim::kPrimMakeTuple)) { in BuildOutput()
505 args.push_back(NewValueNode(prim::kPrimMakeTuple)); in BuildOutput()
517 auto tuple_env = NewCNode({NewValueNode(prim::kPrimMakeTuple), NewValueNode(newenv)}, bprop_fg); in BuildOutput()
520 extra_args.insert(extra_args.begin(), NewValueNode(prim::kPrimMakeTuple)); in BuildOutput()
623 inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in CheckBprop()
721 outputs.push_back(NewValueNode(prim::kPrimMakeTuple)); in FakeBprop()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | order_enforce.cc | 42 } else if (IsPrimitiveCNode(node, prim::kPrimMakeTuple)) { in Run()
72 … if (IsPrimitiveCNode(attach, prim::kPrimLoad) && IsPrimitiveCNode(attach, prim::kPrimMakeTuple)) { in HandleUpdateState()
112 if (IsPrimitiveCNode(user_node, prim::kPrimMakeTuple)) { in FindUpdateStateUsers()
230 if (IsPrimitiveCNode(attach, prim::kPrimMakeTuple)) { in IsInUpdateState()
246 …if (IsPrimitiveCNode(load_user, prim::kPrimMakeTuple) || IsPrimitiveCNode(load_user, prim::kPrimUp… in AddInputEdges()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | zip_operation.cc | 67 make_tuple_nodes.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateFuncGraph()
70 make_tuple_zip_nodes.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateFuncGraph()
|
| D | composite.cc | 201 inputs.push_back(NewValueNode(prim::kPrimMakeTuple)); in FullMake()
428 elems.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateSequeueFuncGraph()
489 params.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateFuncGraph()
506 grads.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateFuncGraph()
516 fg->set_output(fg->NewCNodeInOrder({NewValueNode(prim::kPrimMakeTuple), out, NewValueNode(b)})); in GenerateFuncGraph()
517 (void)fg->transforms().emplace("primal", FuncGraphTransform(prim::kPrimMakeTuple)); in GenerateFuncGraph()
547 grads.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateFuncGraph()
557 fg->set_output(fg->NewCNodeInOrder({NewValueNode(prim::kPrimMakeTuple), out, NewValueNode(b)})); in GenerateFuncGraph()
634 …k_child->set_output(k_child->NewCNodeInOrder({NewValueNode(kPrimMakeTuple), inputs_bprop, fv_bprop… in GradByParameter()
891 elems.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateFuncGraph()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | clean.cc | 221 …auto tuple_new_item = node->func_graph()->NewCNode({NewValueNode(prim::kPrimMakeTuple), item_value… in ConvertDictSetItemToTupleSetItem()
235 inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in ConvertMakeRecordToMakeTuple()
256 return NewValueNode(prim::kPrimMakeTuple); in ErasePartialNode()
262 new_inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in ErasePartialNode()
277 inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in ConvertMakeListToMakeTuple()
400 new_node = NewValueNode(prim::kPrimMakeTuple); in SimplifyDataStructures()
448 inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in ConvertMakeSparseToMakeTuple()
551 std::vector<AnfNodePtr> inputs{NewValueNode(prim::kPrimMakeTuple)}; in ExpandTuplesP()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | anf.cc | 330 } else if (IsPrimitiveCNode(input, prim::kPrimMakeTuple)) { in GetLoadInputs()
413 …} else if (IsPrimitiveCNode(node, prim::kPrimMakeTuple) || IsPrimitiveCNode(node, prim::kPrimUpdat… in GetVirtualNodeTargetFromInputs()
420 …} else if (IsPrimitiveCNode(node, prim::kPrimMakeTuple) && inputs.size() > make_tuple_valid_input_… in GetVirtualNodeTargetFromInputs()
532 … IsPrimitiveCNode(node, prim::kPrimUpdateState) || IsPrimitiveCNode(node, prim::kPrimMakeTuple) || in GetOriginNodeTarget()
538 … IsPrimitiveCNode(node, prim::kPrimUpdateState) || IsPrimitiveCNode(node, prim::kPrimMakeTuple) || in GetOriginNodeTarget()
|
| D | func_graph_extends.cc | 94 var_param_tuple_nodes.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateVarParams()
124 std::vector<AnfNodePtr> kwarg_keys_tuple_nodes = {NewValueNode(prim::kPrimMakeTuple)}; in GenerateKwParams()
125 std::vector<AnfNodePtr> kwarg_values_tuple_nodes = {NewValueNode(prim::kPrimMakeTuple)}; in GenerateKwParams()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | bn_grad_split.cc | 102 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_reduce_grad_ou… in BNGradSplitForTBE()
133 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_reduce_grad_ou… in SyncBNGradSplitForTBE()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_n_optimizer.cc | 50 if (IsPrimitive(input0, prim::kPrimMakeTuple) || IsPrimitive(input0, prim::kPrimTupleGetItem)) { in GetSplitOutputs()
69 if (IsPrimitive(input0, prim::kPrimMakeTuple)) { in VisitSplitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | optimize_dependence.cc | 117 if (AnfAlgo::GetCNodeName(cnode) != prim::kPrimMakeTuple->name()) { in ReplaceMakeTuple()
170 AnfAlgo::CheckPrimitiveType(input, prim::kPrimMakeTuple)) { in SearchTransDataAndCast()
|
| D | getitem_tuple.cc | 41 return VectorRef({prim::kPrimTupleGetItem, VectorRef({prim::kPrimMakeTuple, Xs}), C}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_build.cc | 38 …if (kernel_name == prim::kPrimTupleGetItem->name() || kernel_name == prim::kPrimMakeTuple->name() … in CreateGPUKernel()
|