Home
last modified time | relevance | path

Searched defs:tuple_node (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dconv_tuple_activation_fusion.cc65 AnfNodePtr tuple_node = act_node->input(1); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/model/acl/acl_vm/
Dacl_multi_graph_session.cc141 VectorRef MultiGraphAclSession::ConstructOutputRefByTupleNode(const CNodePtr &tuple_node, in ConstructOutputRefByTupleNode()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dspatial_node_adapter.cc77 auto tuple_node = in_node->cast<CNodePtr>(); in AdapteNodeWithDynamicInput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/
Doperator_info.cc123 auto tuple_node = std::make_shared<ops::TupleGetItem>(); in CreateMultipleOutputsOfAnfNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dshape_unify_mindir.cc183 auto tuple_node = NewValueNode(tuple_value); in CreateScalarValueTuple() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dinference_matmul_split_fusion.cc202 auto tuple_node = reshape_cnode->input(kIndex2); in CreateMatmulSplitNode() local
425 auto tuple_node = reshape_cnode->input(kIndex2); in CreateMatmulBiasAddSplitSiluNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/const_fold/
Dfold_utils.cc171 auto tuple_node = used_node_list->at(0).first; in ReplaceCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dpadv3_ge_pass.cc42 auto tuple_node = NewValueNode(tuple_value); in GenerateDataValueTuple() local
Dcontrol_flow_pass.cc80 …Pass::GetItemVisitedNums(const std::set<AnfNodePtr> &visited_nodes, const AnfNodePtr &tuple_node) { in GetItemVisitedNums()
98 void ControlFlowPass::MoveGetItemToVisited(const size_t &need_size, const AnfNodePtr &tuple_node, in MoveGetItemToVisited()
Dscalar_op_pass.cc217 auto tuple_node = NewValueNode(tuple_value); in GenerateScalarValueTuple() local
Ddecrease_transpose_algo.cc44 auto tuple_node = cnode->input(1)->cast<CNodePtr>(); in __anon90e0e0d90202() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Ddecrease_transfer_precision.cc115 auto tuple_node = in_node->input(1); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_builder.cc281 auto tuple_node = NewFuncNode(value, nullptr, InputType::kOpOutput); in MakeTuple() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_grad.cc788 auto tuple_node = old_cnode->input(kIndex1); in ElimateTupleGetItem() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Danfalgo.cc1004 auto tuple_node = kNodeTupleOutSet.find(node_name); in SetOutputTypeAndDetailShape() local
1080 auto tuple_node = kNodeTupleOutSet.find(node_name); in SetOutputInferTypeAndShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/
Dstatic_analysis.cc1487 const auto &tuple_node = input_node->cast_ptr<CNode>()->input(branch_index); in JoinBranchesFailedInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/
Dpynative_utils.cc334 auto tuple_node = ConvertValueSequenceToMakeTuple(value_node, func_graph); in ConvertValueSequenceToMakeTuple() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dfallback_rewriter.cc568 auto tuple_node = fg->NewCNode({NewValueNode(prim::kPrimMakeTuple), key_node, value_node}); in EraseDictItems() local