Home
last modified time | relevance | path

Searched refs:tuple_get_item (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dconvert_tuple_output_to_maketuple_test.py20 tuple_get_item = Primitive(Constants.kTupleGetItem) variable
48 …make_tuple(tuple_get_item(res, 0), tuple_get_item(res, 1), tuple_get_item(res, 2), tuple_get_item(…
49 tuple_get_item(res, 4)))
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_add_relu_fusion.cc32 VectorRef tuple_get_item = VectorRef({prim::kPrimTupleGetItem, batch_norm, index_}); in DefinePattern() local
33 VectorRef tensor_add = VectorRef({prim::kPrimAdd, tuple_get_item, z_}); in DefinePattern()
45 auto tuple_get_item = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tensor_add), 0); in Process() local
46 MS_EXCEPTION_IF_NULL(tuple_get_item); in Process()
47 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_get_item), 0); in Process()
99 return tuple_get_item; in Process()
Dpost_batch_norm_add_relu_fusion.cc32 VectorRef tuple_get_item = VectorRef({prim::kPrimTupleGetItem, batch_norm, index_}); in DefinePattern() local
33 VectorRef tensor_add = VectorRef({prim::kPrimAdd, z_, tuple_get_item}); in DefinePattern()
45 auto tuple_get_item = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tensor_add), 1); in Process() local
46 MS_EXCEPTION_IF_NULL(tuple_get_item); in Process()
47 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_get_item), 0); in Process()
99 return tuple_get_item; in Process()
Dbatch_norm_relu_fusion.cc42 auto tuple_get_item = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
43 MS_EXCEPTION_IF_NULL(tuple_get_item); in Process()
44 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_get_item), 0); in Process()
94 return tuple_get_item; in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_tensor_move_for_getnext.cc41 auto tuple_get_item = CreatTupleGetItemNode(func_graph, node, output_index); in InsertTensorMoveForGetNextOutputs() local
42 auto new_node = CreateTensorMoveOp(func_graph, tuple_get_item); in InsertTensorMoveForGetNextOutputs()
46 if (AnfAlgo::IsNodeDynamicShape(tuple_get_item)) { in InsertTensorMoveForGetNextOutputs()
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc36 static size_t GetTupleGetItemOutIndex(const mindspore::CNodePtr &tuple_get_item) { in GetTupleGetItemOutIndex() argument
37 MS_ASSERT(tuple_get_item != nullptr); in GetTupleGetItemOutIndex()
38 if (tuple_get_item->size() != mindspore::kTupleGetItemInputSize) { in GetTupleGetItemOutIndex()
42 …auto output_index_value_node = tuple_get_item->input(mindspore::kInputNodeOutputIndexInTupleGetIte… in GetTupleGetItemOutIndex()
/third_party/mindspore/mindspore/lite/tools/optimizer/format/
Dto_format_base.cc151 CNodePtr tuple_get_item = nullptr; in InsertPostTransNode() local
157 tuple_get_item = opt::GenTupleGetItemNode(func_graph, cnode, 0); in InsertPostTransNode()
158 post_node = tuple_get_item; in InsertPostTransNode()
159 manager_->Replace(cnode, tuple_get_item); in InsertPostTransNode()
170 if (tuple_get_item != nullptr) { in InsertPostTransNode()
171 if (!manager_->Replace(tuple_get_item, tuple_get_item->input(1))) { in InsertPostTransNode()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_cond.cc43 auto tuple_get_item = node_user.first; in GetSwitchBranchType() local
44 …if (!utils::isa<CNodePtr>(tuple_get_item) || !CheckPrimitiveType(tuple_get_item, prim::kPrimTupleG… in GetSwitchBranchType()
48 auto tuple_get_item_cnode = utils::cast<CNodePtr>(tuple_get_item); in GetSwitchBranchType()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_tuple_activation_fusion.cc32 auto tuple_get_item = VectorRef({is_tuple_getitem, is_conv, is_var}); in DefinePattern() local
35 return VectorRef({is_activation, tuple_get_item}); in DefinePattern()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Ddecrease_transpose_algo.cc378 CNodePtr tuple_get_item = nullptr; in InsertPostTransNode() local
384 tuple_get_item = GenTupleGetItemNode(func_graph, cnode, 0); in InsertPostTransNode()
385 MS_CHECK_TRUE_RET(tuple_get_item != nullptr, lite::RET_ERROR); in InsertPostTransNode()
386 post_node = tuple_get_item; in InsertPostTransNode()
387 func_graph->manager()->Replace(cnode, tuple_get_item); in InsertPostTransNode()
399 if (tuple_get_item != nullptr) { in InsertPostTransNode()
400 func_graph->manager()->Replace(tuple_get_item, tuple_get_item->input(1)); in InsertPostTransNode()
/third_party/glib/glib/
Dgvarianttypeinfo.c413 tuple_get_item (TupleInfo *info, in tuple_get_item() function
590 while (tuple_get_item (info, items, &d, &e)) in tuple_generate_table()
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.h85 size_t GetTupleGetItemOutIndex(const CNodePtr &tuple_get_item);
Dgllo_utils.cc815 size_t GetTupleGetItemOutIndex(const CNodePtr &tuple_get_item) { in GetTupleGetItemOutIndex() argument
816 if (tuple_get_item == nullptr || tuple_get_item->size() != kInputSizeThree) { in GetTupleGetItemOutIndex()
820 auto output_index_value_node = tuple_get_item->input(kInputIndexTwo); in GetTupleGetItemOutIndex()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc113 auto tuple_get_item = node->cast<CNodePtr>(); in GetRealOutputRecursively() local
114 MS_EXCEPTION_IF_NULL(tuple_get_item); in GetRealOutputRecursively()
115 auto input = AnfAlgo::GetTupleGetItemRealInput(tuple_get_item); in GetRealOutputRecursively()
116 auto index = AnfAlgo::GetTupleGetItemOutIndex(tuple_get_item); in GetRealOutputRecursively()
212 AnfNodePtr AnfRuntimeAlgorithm::GetTupleGetItemRealInput(const CNodePtr &tuple_get_item) { in GetTupleGetItemRealInput() argument
213 MS_EXCEPTION_IF_NULL(tuple_get_item); in GetTupleGetItemRealInput()
214 if (tuple_get_item->size() != kTupleGetItemInputSize) { in GetTupleGetItemRealInput()
217 return tuple_get_item->input(kRealInputNodeIndexInTupleGetItem); in GetTupleGetItemRealInput()
220 size_t AnfRuntimeAlgorithm::GetTupleGetItemOutIndex(const CNodePtr &tuple_get_item) { in GetTupleGetItemOutIndex() argument
221 MS_EXCEPTION_IF_NULL(tuple_get_item); in GetTupleGetItemOutIndex()
[all …]
Danf_runtime_algorithm.h76 static AnfNodePtr GetTupleGetItemRealInput(const CNodePtr &tuple_get_item);
77 static size_t GetTupleGetItemOutIndex(const CNodePtr &tuple_get_item);
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.cc592 auto tuple_get_item = NewValueNode(prim::kPrimTupleGetItem); in GetGrad() local
593 …auto f_app = k_child->NewCNodeInOrder({tuple_get_item, k_app, NewValueNode(static_cast<int64_t>(0)… in GetGrad()
594 …auto bprop = k_child->NewCNodeInOrder({tuple_get_item, k_app, NewValueNode(static_cast<int64_t>(1)… in GetGrad()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc656 auto tuple_get_item = CreatTupleGetItemNode(kernel_graph_ptr, getnext_cnode, 0); in CreateEndOfSequenceOP() local
657 inputs.push_back(tuple_get_item); in CreateEndOfSequenceOP()
667 end_of_sequence_node->set_abstract(tuple_get_item->abstract()); in CreateEndOfSequenceOP()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc84 auto tuple_get_item = kernel_graph->NewCNode(tuple_get_item_inputs); in TEST_F() local
85 kernel_with_index = AnfAlgo::VisitKernel(tuple_get_item, 0); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc315 auto tuple_get_item = func_graph->NewCNode(tuple_get_item_inputs); in InsertMakeTuple() local
316 MS_EXCEPTION_IF_NULL(tuple_get_item); in InsertMakeTuple()
317 make_tuple_inputs.push_back(tuple_get_item); in InsertMakeTuple()