Home
last modified time | relevance | path

Searched refs:kInputNodeOutputIndexInTupleGetItem (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dgetitem_tuple.cc51 AnfNodePtr index_node = tuple_getitem->input(kInputNodeOutputIndexInTupleGetItem); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Deliminate_redundant_output.cc42 …getitem_node->cast<CNodePtr>()->input(kInputNodeOutputIndexInTupleGetItem)->cast<ValueNodePtr>()->… in GetIndex()
52 getitem->set_input(kInputNodeOutputIndexInTupleGetItem, idx_node); in SetIndex()
Drewrite_output_shape.cc71 auto idx = GetValue<int64_t>(GetValueNode(cnode->input(kInputNodeOutputIndexInTupleGetItem))); in Process()
Doptimize_assign.cc114 auto index_node = getitem->cast<CNodePtr>()->input(kInputNodeOutputIndexInTupleGetItem); in GetitemIndex()
Ddecrease_transfer_precision.cc38 auto index_node = getitem->cast<CNodePtr>()->input(kInputNodeOutputIndexInTupleGetItem); in ObtainGetItemIndex()
Dadd_atomic_clean.cc464 auto value_input = get_item_cnode->input(kInputNodeOutputIndexInTupleGetItem); in FindOriginCNodeUsers()
Dgraph_kernel_helper.cc435 auto value_input = get_item_cnode->input(kInputNodeOutputIndexInTupleGetItem); in ReplaceNewFuseCNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dbatchnorm_to_bninfer.cc91 AnfNodePtr index_node = tuple_getitem->input(kInputNodeOutputIndexInTupleGetItem); in NeedFusion()
Dbatchnormgrad_to_bninfergrad.cc93 AnfNodePtr index_node = tuple_getitem->input(kInputNodeOutputIndexInTupleGetItem); in NeedFusion()
Dfused_batch_norm_fusion.cc226 AnfNodePtr index_node = tuple_getitem_cnode->input(kInputNodeOutputIndexInTupleGetItem); in Process()
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc42 …to output_index_value_node = tuple_get_item->input(mindspore::kInputNodeOutputIndexInTupleGetItem); in GetTupleGetItemOutIndex()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbatch_norm_bert_fission.cc48 auto index_node = tuple_getiterm_cnode->input(kInputNodeOutputIndexInTupleGetItem); in GetBatchNormOutputs()
Dbatch_norm_grad_infer_fission.cc43 auto index_node = tuple_getiterm_cnode->input(kInputNodeOutputIndexInTupleGetItem); in CheckOutputsIndex()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dsplit_n_optimizer.cc77 auto input2 = cnode->input(kInputNodeOutputIndexInTupleGetItem); in VisitSplitKernel()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_add_relu_grad_fusion.cc54 auto index_node = tuple_getiterm_cnode->input(kInputNodeOutputIndexInTupleGetItem); in GetBatchNormOutputs()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc863 auto index_input1 = n1_cnode->input(kInputNodeOutputIndexInTupleGetItem); in CompareTupleGetitem()
867 auto index_input2 = n2_cnode->input(kInputNodeOutputIndexInTupleGetItem); in CompareTupleGetitem()
1009 auto getitem_input2 = out_getitem_ptr->input(kInputNodeOutputIndexInTupleGetItem); in GetNodeOutputUsedNum()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h511 constexpr auto kInputNodeOutputIndexInTupleGetItem = 2; variable
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc225 auto output_index_value_node = tuple_get_item->input(kInputNodeOutputIndexInTupleGetItem); in GetTupleGetItemOutIndex()
254 auto input2 = cnode->input(kInputNodeOutputIndexInTupleGetItem); in VisitKernel()