Home
last modified time | relevance | path

Searched refs:kPrimTupleGetItem (Results 1 – 25 of 130) sorted by relevance

123456

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dless_batch_normalization.cc41 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu},
44 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D},
57 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu},
71 …{prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu, prim::kPrimTup…
72 …prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu, prim::kPrimTupleGetItem, prim::kPrimBatc…
76 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D},
87 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu},
90 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D},
101 …{prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu, prim::kPrimTup…
114 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimRelu},
[all …]
Ditem_tuple_or_list_eliminate.h45 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsVNode})(node); in operator()
103 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsVNode})(node); in operator()
151 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsVNode, IsVNode})(node); in operator()
299 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsVNode})(node); in operator()
307 return fg->NewCNode({NewValueNode(prim::kPrimTupleGetItem), tuple_, c2_}); in operator()
369 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
388 auto item_node = NewCNode({NewValueNode(prim::kPrimTupleGetItem), x_, c_}, fg); in operator()
Dbranch_culling.h61 … PPrimitive(prim::kPrimTupleGetItem, PPrimitive(prim::kPrimSwitch, cond, true_br, false_br), x), in operator()
62 … PPrimitive(prim::kPrimSwitch, cond, PPrimitive(prim::kPrimTupleGetItem, true_br, x), in operator()
63 PPrimitive(prim::kPrimTupleGetItem, false_br, x)), in operator()
Dminmax_grad.h57 …AnfVisitor::Match(prim::kPrimTupleGetItem, {internal::IsOriginMaxMinGrad, IsValueNode<Int64Imm>})(… in operator()
91 …return fg->NewCNode({NewValueNode(prim::kPrimTupleGetItem), tuple, NewValueNode(MakeValue(idx_))}); in operator()
Dincorporate_getitem.h67 … auto output_node = new_fg->NewCNode({NewValueNode(prim::kPrimTupleGetItem), output, idx_node}); in operator()
119 … auto output_node = new_fg->NewCNode({NewValueNode(prim::kPrimTupleGetItem), output, idx_node}); in operator()
331 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
388 if (IsPrimitiveCNode(user.first, prim::kPrimTupleGetItem)) { in MultipleUse()
435 …node->func_graph()->NewCNode({NewValueNode(prim::kPrimTupleGetItem), fg_call_cnode_, NewValueNode(… in ShrinkFuncGraphOutput()
451 …node->func_graph()->NewCNode({NewValueNode(prim::kPrimTupleGetItem), fg_call_cnode_, NewValueNode(… in ShrinkFuncGraphOutput()
535 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
621 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
719 …node->func_graph()->NewCNode({NewValueNode(prim::kPrimTupleGetItem), switch_call_cnode, NewValueNo… in ShrinkFuncGraphOutput()
740 …node->func_graph()->NewCNode({NewValueNode(prim::kPrimTupleGetItem), switch_call_cnode, NewValueNo… in ShrinkFuncGraphOutput()
[all …]
Dbranch_culling.cc38 std::vector<AnfNodePtr> tuple_getitem_nodes{NewValueNode(prim::kPrimTupleGetItem), switch_apply, in GenerateSwitchNode()
60 … {prim::kPrimTupleGetItem, {2}}, in InConvertWhiteList()
85 {prim::kPrimStateSetItem, {1}}, {prim::kPrimTupleGetItem, {2}}, in InConvertWhiteList()
487 std::vector<AnfNodePtr> tuple_getitem_nodes{NewValueNode(prim::kPrimTupleGetItem), in GenerateMergeNodes()
498 … std::vector<AnfNodePtr> true_getitem_nodes{NewValueNode(prim::kPrimTupleGetItem), block_nodes[1], in GenerateMergeNodes()
501 … std::vector<AnfNodePtr> false_getitem_nodes{NewValueNode(prim::kPrimTupleGetItem), block_nodes[2], in GenerateMergeNodes()
Drecompute_prepare.h63 …Node(output_cnode, prim::kPrimDepend) || IsPrimitiveCNode(output_cnode, prim::kPrimTupleGetItem)) { in GetRealOutputNodes()
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dtf_bidirection_gru_cf_fusion.cc66 auto rt = VectorRef({GetPrim(prim::kPrimTupleGetItem), split, std::make_shared<Var>()}); in DefineGruCellPattern()
67 auto zt = VectorRef({GetPrim(prim::kPrimTupleGetItem), split, std::make_shared<Var>()}); in DefineGruCellPattern()
97 …auto fw_switch_true = VectorRef({GetPrim(prim::kPrimTupleGetItem), fw_switch, std::make_shared<Var… in DefineBidirectionRnnPattern()
122 …VectorRef({GetPrim(prim::kPrimTupleGetItem), counter_switch1, std::make_shared<Var>()}); // ident… in DefineBidirectionRnnPattern()
125 auto in_ta_handle = VectorRef({GetPrim(prim::kPrimTupleGetItem), in_ta, std::make_shared<Var>()}); in DefineBidirectionRnnPattern()
126 auto in_ta_flow = VectorRef({GetPrim(prim::kPrimTupleGetItem), in_ta, std::make_shared<Var>()}); in DefineBidirectionRnnPattern()
141 …auto switch3_true = VectorRef({GetPrim(prim::kPrimTupleGetItem), switch3, std::make_shared<Var>()}… in DefineBidirectionRnnPattern()
146 …auto out_ta_handle = VectorRef({GetPrim(prim::kPrimTupleGetItem), out_ta, std::make_shared<Var>()}… in DefineBidirectionRnnPattern()
147 auto out_ta_flow = VectorRef({GetPrim(prim::kPrimTupleGetItem), out_ta, std::make_shared<Var>()}); in DefineBidirectionRnnPattern()
150 …auto switch2_true = VectorRef({GetPrim(prim::kPrimTupleGetItem), std::make_shared<SeqVar>()}); //… in DefineBidirectionRnnPattern()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dirpass.cc77 {prim::kPrimTupleGetItem, prim::kPrimListGetItem}); in OptimizeIRPassLib()
80 {prim::kPrimTupleGetItem, prim::kPrimListGetItem}); in OptimizeIRPassLib()
86 {prim::kPrimTupleGetItem, prim::kPrimListGetItem}); in OptimizeIRPassLib()
89 {prim::kPrimTupleGetItem, prim::kPrimListGetItem}); in OptimizeIRPassLib()
92 …{prim::kPrimTupleGetItem, prim::kPrimTupleSetItem, prim::kPrimListGetItem, prim::kPrimListSetItem}… in OptimizeIRPassLib()
151 …= MakeSubstitution(std::make_shared<MinMaximumGrad>(), "minmaximum_grad", prim::kPrimTupleGetItem); in OptimizeIRPassLib()
156 … "float_tuple_getitem_switch", prim::kPrimTupleGetItem); in OptimizeIRPassLib()
206 …ion(std::make_shared<IncorporateGetitemSet>(), "incorporate_getitem_set", prim::kPrimTupleGetItem); in OptimizeIRPassLib()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dfused_batch_norm_fusion.cc221 if (!IsPrimitiveCNode(output, prim::kPrimTupleGetItem)) { in Process()
250 VectorRef tuple_getitem0 = VectorRef({prim::kPrimTupleGetItem, batch_norm, index0}); in DefinePattern()
251 VectorRef tuple_getitem1 = VectorRef({prim::kPrimTupleGetItem, batch_norm, index1}); in DefinePattern()
252 VectorRef tuple_getitem2 = VectorRef({prim::kPrimTupleGetItem, batch_norm, index2}); in DefinePattern()
269 VectorRef tuple_getitem0 = VectorRef({prim::kPrimTupleGetItem, batch_norm, index0}); in DefinePattern()
270 VectorRef tuple_getitem1 = VectorRef({prim::kPrimTupleGetItem, batch_norm, index1}); in DefinePattern()
271 VectorRef tuple_getitem2 = VectorRef({prim::kPrimTupleGetItem, batch_norm, index2}); in DefinePattern()
292 VectorRef tuple_getitem0 = VectorRef({prim::kPrimTupleGetItem, batch_norm, index0}); in DefinePattern()
293 VectorRef tuple_getitem1 = VectorRef({prim::kPrimTupleGetItem, batch_norm, index1}); in DefinePattern()
294 VectorRef tuple_getitem2 = VectorRef({prim::kPrimTupleGetItem, batch_norm, index2}); in DefinePattern()
Dlayer_norm_beta_gamma_backprop_fusion.cc37 if (AnfAlgo::GetCNodeName(output_cnode) != prim::kPrimTupleGetItem->name()) { in GetOutputCastNodes()
39 … << prim::kPrimTupleGetItem->name() << " trace: " << trace::DumpSourceLines(node); in GetOutputCastNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dadd_training_attr.cc41 if (AnfAlgo::CheckPrimitiveType(output, prim::kPrimTupleGetItem)) { in CheckOP()
73 …= nullptr || func_graph == nullptr || AnfAlgo::CheckPrimitiveType(node, prim::kPrimTupleGetItem) || in Process()
Dgetitem_tuple.cc41 return VectorRef({prim::kPrimTupleGetItem, VectorRef({prim::kPrimMakeTuple, Xs}), C}); in DefinePattern()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.cc214 … {NewValueNode(prim::kPrimTupleGetItem), item.first, NewValueNode(SizeToLong(pos))}); in FullMake()
429 op = prim::kPrimTupleGetItem; in GenerateSequeueFuncGraph()
509 …grads.push_back(b->NewCNodeInOrder({NewValueNode(prim::kPrimTupleGetItem), dout, NewValueNode(i)})… in GenerateFuncGraph()
592 auto tuple_get_item = NewValueNode(prim::kPrimTupleGetItem); in GetGrad()
619 …k_child->NewCNodeInOrder({NewValueNode(prim::kPrimTupleGetItem), b_app, NewValueNode(static_cast<i… in GradByParameter()
756 return fg_ptr->NewCNodeInOrder({NewValueNode(prim::kPrimTupleGetItem), item}); in GenerateFuncGraph()
761 {NewValueNode(prim::kPrimTupleGetItem), item, NewValueNode(static_cast<int64_t>(1))}); in GenerateFuncGraph()
840 return fg_ptr->NewCNodeInOrder({NewValueNode(prim::kPrimTupleGetItem), item, nullptr}); in MakeNext()
846 {NewValueNode(prim::kPrimTupleGetItem), item, NewValueNode(static_cast<int64_t>(1))}); in MakeNext()
895 …elems.push_back(ret->NewCNodeInOrder({NewValueNode(prim::kPrimTupleGetItem), p_tup_a, NewValueNode… in GenerateFuncGraph()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_tensor_move_for_cascade.cc31 if (!AnfAlgo::CheckPrimitiveType(node, prim::kPrimTupleGetItem)) { in IsPartOutputsOfHcclOp()
55 if (IsPrimitiveCNode(output, prim::kPrimTupleGetItem)) { in IsPartOutputsOfHcclOp()
Dsplit_n_optimizer.cc50 if (IsPrimitive(input0, prim::kPrimMakeTuple) || IsPrimitive(input0, prim::kPrimTupleGetItem)) { in GetSplitOutputs()
73 } else if (IsPrimitive(input0, prim::kPrimTupleGetItem)) { in VisitSplitKernel()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dglu_fusion.cc62 auto is_tuple_getitem1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTupleGetItem>); in DefinePattern()
67 auto is_tuple_getitem2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTupleGetItem>); in DefinePattern()
Dconv_tuplegetitem_fusion.cc24 auto is_tuple_getitem = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTupleGetItem>); in DefinePattern()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Deliminate_concat_split.cc50 } else if (IsPrimitive(input0, prim::kPrimTupleGetItem)) { in GetRealPrevCNode()
106 if (!CheckPrimitiveType(tmp_cnode, prim::kPrimTupleGetItem)) { in ConcatSplitEliminate()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dreduce_precision_fusion.cc63 if (AnfAlgo::GetCNodeName(used_node) == prim::kPrimTupleGetItem->name()) { in ProcessTupleGetItem()
98 if (AnfAlgo::GetCNodeName(used_node) == prim::kPrimTupleGetItem->name()) { in Run()
Dinsert_cast_gpu.cc59 if (AnfAlgo::GetCNodeName(used_node) == prim::kPrimTupleGetItem->name()) { in ProcessTupleGetItem()
101 if (AnfAlgo::GetCNodeName(used_node) == prim::kPrimTupleGetItem->name()) { in Run()
Dinsert_format_transform_op.cc160 if (AnfAlgo::GetCNodeName(used_node) == prim::kPrimTupleGetItem->name()) { in Process()
182 if (AnfAlgo::GetCNodeName(used_node) == prim::kPrimTupleGetItem->name()) { in ProcessForTupleItem()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/
Dinsert_format_transform_op.cc96 if (AnfAlgo::GetCNodeName(used_node) == prim::kPrimTupleGetItem->name()) { in ProcessForTupleItem()
153 if (AnfAlgo::GetCNodeName(used_node) == prim::kPrimTupleGetItem->name()) { in InsertTransformOpForOutput()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_build.cc38 …if (kernel_name == prim::kPrimTupleGetItem->name() || kernel_name == prim::kPrimMakeTuple->name() … in CreateGPUKernel()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_cluster.cc144 if (IsPrimitiveCNode(node, prim::kPrimTupleGetItem)) { in Cluster()
368 if (!IsClusterableOp(node) && !IsPrimitiveCNode(node, prim::kPrimTupleGetItem)) { in FindCandidates()
402 return !IsPrimitiveCNode(this->nodes_[node_id], prim::kPrimTupleGetItem); in Process()
464 if (IsPrimitiveCNode(nodes_[cluster_id], prim::kPrimTupleGetItem)) { in RemoveWildGetitem()

123456