Home
last modified time | relevance | path

Searched refs:tuple_getitem (Results 1 – 25 of 76) sorted by relevance

1234

/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dsplit_fission_test.py22 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
48 splitv1 = splitv(tuple_getitem(splitv0, 0))
49 splitv2 = splitv(tuple_getitem(splitv0, 1))
50 splitv3 = splitv(tuple_getitem(splitv0, 2))
51 …make_tuple0 = make_tuple(tuple_getitem(splitv1, 0), tuple_getitem(splitv1, 1), tuple_getitem(split…
52 … tuple_getitem(splitv2, 0), tuple_getitem(splitv2, 1), tuple_getitem(splitv2, 2),
53 tuple_getitem(splitv3, 0), tuple_getitem(splitv3, 1))
55 …make_tuple(tuple_getitem(make_tuple0, 0), tuple_getitem(make_tuple0, 1), tuple_getitem(make_tuple0…
56 … tuple_getitem(make_tuple0, 3), tuple_getitem(make_tuple0, 4), tuple_getitem(make_tuple0, 5),
57 tuple_getitem(make_tuple0, 6), tuple_getitem(make_tuple0, 7)))
Dbn_grad_split.py23 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
54 item0 = tuple_getitem(bn_grad_output, 0)
55 item1 = tuple_getitem(bn_grad_output, 1)
56 item2 = tuple_getitem(bn_grad_output, 2)
63 bn_grad1_item0 = tuple_getitem(bn_grad1_output, 0)
64 bn_grad1_item1 = tuple_getitem(bn_grad1_output, 1)
65 bn_grad1_item2 = tuple_getitem(bn_grad1_output, 2)
67 bn_grad2_item0 = tuple_getitem(bn_grad2_output, 0)
68 bn_grad2_item1 = tuple_getitem(bn_grad2_output, 1)
69 bn_grad2_item2 = tuple_getitem(bn_grad2_output, 2)
[all …]
Dbatch_norm_grad_infer_fission_test.py20 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
44 …outputs = make_tuple(tuple_getitem(batch_norm, 0), tuple_getitem(batch_norm, 1), tuple_getitem(bat…
45 output = tuple_getitem(outputs, 0)
51 …outputs = make_tuple(tuple_getitem(batch_norm, 0), tuple_getitem(batch_norm, 1), tuple_getitem(bat…
52 output = tuple_getitem(outputs, 0)
58 …outputs = make_tuple(tuple_getitem(batch_norm, 0), tuple_getitem(batch_norm, 1), tuple_getitem(bat…
59 output = tuple_getitem(outputs, 0)
66 outputs = make_tuple(bn_infer_grad, tuple_getitem(bn_training_update_grad, 0),
67 tuple_getitem(bn_training_update_grad, 1))
68 …new_outputs = make_tuple(tuple_getitem(outputs, 0), tuple_getitem(outputs, 1), tuple_getitem(outpu…
[all …]
Dfused_batch_norm_fusion_test.py27 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
55 sub0 = Sub(var0, tuple_getitem(batch_norm, 1))
56 sub1 = Sub(var1, tuple_getitem(batch_norm, 2))
63 depend0 = F.depend(tuple_getitem(batch_norm, 0), assign_sub0)
65 … outputs = make_tuple(depend1, tuple_getitem(batch_norm, 3), tuple_getitem(batch_norm, 4), u1)
66 output = tuple_getitem(outputs, 0)
72 sub0 = Sub(Cast(var0, mstype.float32), tuple_getitem(batch_norm, 1))
73 sub1 = Sub(Cast(var1, mstype.float32), tuple_getitem(batch_norm, 2))
80 depend0 = F.depend(tuple_getitem(batch_norm, 0), assign_sub0)
82 … outputs = make_tuple(depend1, tuple_getitem(batch_norm, 3), tuple_getitem(batch_norm, 4), u1)
[all …]
Dbn_split.py24 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
55 output = tuple_getitem(bn_output, 0)
61 fused_bn2_input0 = tuple_getitem(fused_bn1_output, 0)
62 fused_bn2_input1 = tuple_getitem(fused_bn1_output, 1)
64 fused_bn3_input1 = tuple_getitem(fused_bn1_output, 0)
65 fused_bn3_input2 = tuple_getitem(fused_bn2_output, 0)
67 output1 = tuple_getitem(fused_bn2_output, 1)
68 output2 = tuple_getitem(fused_bn2_output, 2)
69 output3 = tuple_getitem(fused_bn1_output, 0)
70 output4 = tuple_getitem(fused_bn2_output, 0)
[all …]
Dsingle_batch_norm_fission_test.py20 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
43 item0 = tuple_getitem(batch_norm, 0)
44 item1 = tuple_getitem(batch_norm, 1)
45 item2 = tuple_getitem(batch_norm, 2)
46 item3 = tuple_getitem(batch_norm, 3)
47 item4 = tuple_getitem(batch_norm, 4)
54 reduce_item0 = tuple_getitem(reduce, 0)
55 reduce_item1 = tuple_getitem(reduce, 1)
57 update_item0 = tuple_getitem(update, 0)
58 update_item1 = tuple_getitem(update, 1)
[all …]
Dir_fusion_test.py21 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
66 y1 = tuple_getitem(ar, 0)
67 y2 = tuple_getitem(ar, 1)
68 y3 = tuple_getitem(ar, 2)
69 y4 = tuple_getitem(ar, 3)
70 y5 = tuple_getitem(ar, 4)
94 y1 = tuple_getitem(ar1, 0)
95 y2 = tuple_getitem(ar1, 1)
96 y3 = tuple_getitem(ar2, 0)
97 y4 = tuple_getitem(ar2, 1)
[all …]
Dbatch_norm_bert_fission_test.py20 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
43 …outputs = make_tuple(tuple_getitem(batch_norm, 0), tuple_getitem(batch_norm, 3), tuple_getitem(bat…
44 output = tuple_getitem(outputs, 0)
50 bn_training_update_v2 = BNTrainingUpdateV2(input0, tuple_getitem(bn_training_reduce, 0),
51 … tuple_getitem(bn_training_reduce, 1), input1, input2)
52 …outputs = make_tuple(tuple_getitem(bn_training_update_v2, 0), tuple_getitem(bn_training_update_v2,…
53 tuple_getitem(bn_training_update_v2, 2))
54 output = tuple_getitem(outputs, 0)
Dlayer_norm_grad_split.py21 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
45 item0 = tuple_getitem(layer_norm_grad_output, 0)
46 item1 = tuple_getitem(layer_norm_grad_output, 1)
47 item2 = tuple_getitem(layer_norm_grad_output, 2)
54 x_item0 = tuple_getitem(layer_norm_x_output, 0)
55 x_item1 = tuple_getitem(layer_norm_x_output, 1)
57 beta_item0 = tuple_getitem(layer_norm_beta_output, 0)
58 beta_item1 = tuple_getitem(layer_norm_beta_output, 1)
60 item0 = tuple_getitem(mt, 0)
61 item1 = tuple_getitem(mt, 1)
[all …]
Dremove_internal_output_test.py19 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
65 res = make_tuple(tuple_getitem(max_pool_res, 0), tuple_getitem(max_pool_res, 1))
71 trans_data0 = trans_data(tuple_getitem(output, 0))
72 trans_data1 = trans_data(tuple_getitem(output, 1))
74 res = make_tuple(tuple_getitem(new_make_tuple, 0), tuple_getitem(new_make_tuple, 1))
80 new_make_tuple = make_tuple(tuple_getitem(output, 0), tuple_getitem(output, 1))
81 res = make_tuple(tuple_getitem(new_make_tuple, 0), tuple_getitem(new_make_tuple, 1))
Dbatch_norm_grad_split.py24 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
44 item0 = tuple_getitem(bn_grad_output, 0)
45 item1 = tuple_getitem(bn_grad_output, 1)
46 item2 = tuple_getitem(bn_grad_output, 2)
53 update_item0 = tuple_getitem(bn_update_grad_output, 0)
54 update_item1 = tuple_getitem(bn_update_grad_output, 1)
57 item0 = tuple_getitem(output, 0)
58 item1 = tuple_getitem(output, 1)
59 item2 = tuple_getitem(output, 2)
Dlamb_next_mv_rule_test.py25 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
61 output = tuple_getitem(outputs, 0)
70 outputs = make_tuple(tuple_getitem(lamb_next_mv, 0), tuple_getitem(lamb_next_mv, 1),
71 tuple_getitem(lamb_next_mv, 2), tuple_getitem(lamb_next_mv, 3))
72 output = tuple_getitem(outputs, 0)
96 output = tuple_getitem(outputs, 0)
120 output = tuple_getitem(outputs, 0)
144 output = tuple_getitem(outputs, 0)
168 output = tuple_getitem(outputs, 0)
196 output = tuple_getitem(outputs, 0)
[all …]
Dlamb_next_mv_with_decay_rule_test.py25 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
62 output = tuple_getitem(outputs, 0)
72 …outputs = make_tuple(tuple_getitem(lamb_next_mv_with_decay, 0), tuple_getitem(lamb_next_mv_with_de…
73 … tuple_getitem(lamb_next_mv_with_decay, 2), tuple_getitem(lamb_next_mv_with_decay, 3))
74 output = tuple_getitem(outputs, 0)
99 output = tuple_getitem(outputs, 0)
124 output = tuple_getitem(outputs, 0)
149 output = tuple_getitem(outputs, 0)
174 output = tuple_getitem(outputs, 0)
203 output = tuple_getitem(outputs, 0)
[all …]
Dlayer_norm_beta_gamma_backprop_fusion_test.py22 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
42 output0 = Cast(tuple_getitem(layer, 0))
43 output1 = Cast(tuple_getitem(layer, 1))
50 output0 = Cast(tuple_getitem(layer, 0))
51 output1 = Cast(tuple_getitem(layer, 1))
64 output0 = tuple_getitem(layer, 0)
65 output1 = tuple_getitem(layer, 1)
Dadam_apply_one_fusion_test.py29 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
63 output = tuple_getitem(outputs, 0)
81 output = tuple_getitem(outputs, 0)
99 output = tuple_getitem(outputs, 0)
117 output = tuple_getitem(outputs, 0)
135 output = tuple_getitem(outputs, 0)
141 outputs = make_tuple(tuple_getitem(adam_apply_one, 0), tuple_getitem(adam_apply_one, 1),
142 tuple_getitem(adam_apply_one, 2))
143 output = tuple_getitem(outputs, 0)
173 output = tuple_getitem(outputs, 0)
[all …]
Dlamb_next_right_rule_test.py24 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
52 output = tuple_getitem(outputs, 0)
65 output = tuple_getitem(outputs, 0)
71 outputs = make_tuple(tuple_getitem(lamb_next_right, 0), tuple_getitem(lamb_next_right, 1))
72 output = tuple_getitem(outputs, 0)
Dinsert_tensor_move_for_getnext.py24 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
49 data = tuple_getitem(res, 0)
50 label = tuple_getitem(res, 1)
54 get_item0 = tuple_getitem(bind_tuple, 0)
55 get_item1 = tuple_getitem(bind_tuple, 1)
Dbatchnorm_to_bninfer.py23 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
43 res = tuple_getitem(res, 0)
54 item0 = tuple_getitem(res, 0)
55 item1 = tuple_getitem(res, 1)
56 item2 = tuple_getitem(res, 2)
Dbatchnormgrad_to_bninfergrad.py23 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
43 res = tuple_getitem(res, 0)
54 item0 = tuple_getitem(res, 0)
55 item1 = tuple_getitem(res, 1)
56 item2 = tuple_getitem(res, 2)
Dinsert_trans_op_test.py19 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
63 res = tuple_getitem(max_pool_res, 0)
69 transdata0 = transdata(tuple_getitem(output, 0))
70 transdata1 = transdata(tuple_getitem(output, 1))
72 res = tuple_getitem(res, 0)
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dgetitem_tuple.cc46 auto tuple_getitem = node->cast<CNodePtr>(); in Process() local
47 MS_EXCEPTION_IF_NULL(tuple_getitem); in Process()
48 CheckCNodeInputSize(tuple_getitem, kTupleGetItemInputTensorNum); in Process()
49 AnfNodePtr make_tuple_anf = tuple_getitem->input(kRealInputNodeIndexInTupleGetItem); in Process()
51 AnfNodePtr index_node = tuple_getitem->input(kInputNodeOutputIndexInTupleGetItem); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dsplit_fission.cc77 auto tuple_getitem = func_graph->NewCNode({NewValueNode(prim::kPrimTupleGetItem), input, idx}); in CreateTupleGetItem() local
78 return tuple_getitem; in CreateTupleGetItem()
146 auto tuple_getitem = CreateTupleGetItem(func_graph, base_splitv, nodes_num); in DoFission() local
147 base_splitv_outputs.push_back(tuple_getitem); in DoFission()
148 CNodePtr new_splitv = CreateSplitVNode(func_graph, tuple_getitem); in DoFission()
159 auto tuple_getitem = CreateTupleGetItem(func_graph, base_splitv, nodes_num); in DoFission() local
160 base_splitv_outputs.push_back(tuple_getitem); in DoFission()
161 CNodePtr new_splitv = CreateSplitVNode(func_graph, tuple_getitem); in DoFission()
173 auto tuple_getitem = CreateTupleGetItem(func_graph, base_splitv, nodes_num); in DoFission() local
174 base_splitv_outputs.push_back(tuple_getitem); in DoFission()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dbnupdate_eltwise_eltwise_fusion_pass.cc39 auto tuple_getitem = add->input(kIndex1); in MatchBnupdateAddRelu() local
40 MS_EXCEPTION_IF_NULL(tuple_getitem); in MatchBnupdateAddRelu()
41 …if (tuple_getitem->isa<CNode>() && AnfAlgo::GetCNodeName(tuple_getitem) == prim::kPrimTupleGetItem… in MatchBnupdateAddRelu()
42 auto getitem = tuple_getitem->cast<CNodePtr>(); in MatchBnupdateAddRelu()
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/optimizer/
Dopt_test.py32 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
586 return tuple_getitem(make_tuple(x, y), 0)
590 return tuple_getitem(make_tuple(x, y), 1)
635 return tuple_getitem(tuple_setitem(t, 0, x), 0)
643 return tuple_getitem(tuple_setitem(t, 0, x), 1)
647 return tuple_getitem(t, 1)
738 return tuple_getitem(fn(x, y), 0)
752 return tuple_getitem(fn(x, y), 0)
757 return tuple_getitem(x, 0)
777 return tuple_getitem(
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dbn_grad_split_test.cc65 auto tuple_getitem = make_tuple2->input(1)->cast<CNodePtr>(); in TEST_F() local
66 EXPECT_NE(tuple_getitem->input(1), nullptr); in TEST_F()
67 EXPECT_TRUE(tuple_getitem->input(1)->isa<CNode>()); in TEST_F()
68 auto bn_grad = tuple_getitem->input(1)->cast<CNodePtr>(); in TEST_F()
131 auto tuple_getitem = make_tuple2->input(1)->cast<CNodePtr>(); in TEST_F() local
132 EXPECT_NE(tuple_getitem->input(1), nullptr); in TEST_F()
133 EXPECT_TRUE(tuple_getitem->input(1)->isa<CNode>()); in TEST_F()
134 auto bn_grad = tuple_getitem->input(1)->cast<CNodePtr>(); in TEST_F()

1234