Home
last modified time | relevance | path

Searched defs:tuple_get_item (Results 1 – 17 of 17) sorted by relevance

/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
45 auto tuple_get_item = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tensor_add), 0); in Process() local
Dpost_batch_norm_add_relu_fusion.cc32 VectorRef tuple_get_item = VectorRef({prim::kPrimTupleGetItem, batch_norm, index_}); in DefinePattern() local
45 auto tuple_get_item = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tensor_add), 1); in Process() local
Dbatch_norm_relu_fusion.cc42 auto tuple_get_item = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
/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
/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
/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
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc36 static size_t GetTupleGetItemOutIndex(const mindspore::CNodePtr &tuple_get_item) { in GetTupleGetItemOutIndex()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_cond.cc43 auto tuple_get_item = node_user.first; in GetSwitchBranchType() local
/third_party/mindspore/mindspore/lite/tools/optimizer/format/
Dto_format_base.cc151 CNodePtr tuple_get_item = nullptr; in InsertPostTransNode() local
/third_party/glib/glib/
Dgvarianttypeinfo.c413 tuple_get_item (TupleInfo *info, in tuple_get_item() function
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Ddecrease_transpose_algo.cc378 CNodePtr tuple_get_item = nullptr; in InsertPostTransNode() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc113 auto tuple_get_item = node->cast<CNodePtr>(); in GetRealOutputRecursively() local
212 AnfNodePtr AnfRuntimeAlgorithm::GetTupleGetItemRealInput(const CNodePtr &tuple_get_item) { in GetTupleGetItemRealInput()
220 size_t AnfRuntimeAlgorithm::GetTupleGetItemOutIndex(const CNodePtr &tuple_get_item) { in GetTupleGetItemOutIndex()
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc815 size_t GetTupleGetItemOutIndex(const CNodePtr &tuple_get_item) { in GetTupleGetItemOutIndex()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.cc592 auto tuple_get_item = NewValueNode(prim::kPrimTupleGetItem); in GetGrad() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc656 auto tuple_get_item = CreatTupleGetItemNode(kernel_graph_ptr, getnext_cnode, 0); in CreateEndOfSequenceOP() local
/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
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc315 auto tuple_get_item = func_graph->NewCNode(tuple_get_item_inputs); in InsertMakeTuple() local