Home
last modified time | relevance | path

Searched defs:tuple_getitem (Results 1 – 25 of 75) sorted by relevance

123

/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
146 auto tuple_getitem = CreateTupleGetItem(func_graph, base_splitv, nodes_num); in DoFission() local
159 auto tuple_getitem = CreateTupleGetItem(func_graph, base_splitv, nodes_num); in DoFission() local
173 auto tuple_getitem = CreateTupleGetItem(func_graph, base_splitv, nodes_num); in DoFission() local
/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
131 auto tuple_getitem = make_tuple2->input(1)->cast<CNodePtr>(); in TEST_F() local
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dtopk_split_test.py21 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dgetitem_tuple.py20 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dinsert_tensor_move_for_getnext.py24 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dsplit_fission_test.py22 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dreduce_min_fission_test.py21 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dtranspose_transdata_fusion_test.py19 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dderelu_fusion.py24 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dtensor_scatter_update_fission_test.py23 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dconfusion_softmax_grad_rule.py24 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dlars_v2_fission_test.py23 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dsquare_sum_fusion.py20 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dconfusion_mul_grad_fusion.py24 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dbatch_norm_bert_fission_test.py20 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dclip_by_norm_no_div_square_sum_fusion.py25 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dclip_by_value_fusion.py23 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dsingle_batch_norm_fission_test.py20 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dmul_addn_fusion_test.py25 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dinsert_trans_op_test.py19 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
Dmul_add_fusion_test.py23 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
/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
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dgetitem_tuple.cc46 auto tuple_getitem = node->cast<CNodePtr>(); in Process() local
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/session/
Dsession_test.py24 tuple_getitem = Primitive(Constants.kTupleGetItem) variable
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dremove_internal_output.cc77 auto tuple_getitem = input_node->cast<CNodePtr>(); in Process() local

123