Home
last modified time | relevance | path

Searched defs:tuple_get_item_node (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dadd_cast_rms_norm_cast_fusion.cc100 auto tuple_get_item_node = common::AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_splitter.cc110 CNodePtr tuple_get_item_node = func_graph->NewCNode(tuple_get_item_inputs); in CreateTupleGetItemNode() local
237 auto tuple_get_item_node = CreateTupleGetItemNode(func_graph, src_node, kIndex0); in CreateSendNode() local
808 auto &tuple_get_item_node = make_tuple_get_item_nodes.second; in ProcessForSplitOptimizer() local
891 CNodePtr tuple_get_item_node = CreateTupleGetItemNode(func_graph_, make_tuple_node, kIndex0); in CreateNodesForMakeTuple() local
1417 OperatorLabel GraphSplitter::RecursiveSetTupeGetItemLabel(const CNodePtr &tuple_get_item_node) { in RecursiveSetTupeGetItemLabel()
2127 …CNodePtr tuple_get_item_node = CreateTupleGetItemNode(func_graph_, fused_recv_node, IntToSize(outp… in ReplaceOriginNodesWithRecv() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dsplit_layernorm_comm_fp.cc251 auto tuple_get_item_node = node_user.first; in ExpandSliceRangeToLeft() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.cc762 auto tuple_get_item_node = tensor_list_stack_cnode->input(1); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc1597 const CNodePtr &tuple_get_item_node) { in UpdateUpdateStateForMergeConcatSlice()
1666 auto tuple_get_item_node = call_cnode->func_graph()->NewCNode(tuple_get_item_inputs); in HandleFuncConcatSlice() local