Home
last modified time | relevance | path

Searched defs:sub_graph (Results 1 – 25 of 75) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mindir_loader/mindir_model/
Dmindir_model_loader.cc74 auto *sub_graph = new LiteGraph::SubGraph(); in ConvertModel() local
85 auto *sub_graph = new LiteGraph::SubGraph(); in ConvertModel() local
138 …irModelLoader::ConvertGraph(const mind_ir::GraphProto &graph_proto, LiteGraph::SubGraph *sub_graph, in ConvertGraph()
149 …ModelLoader::ConvertTensors(const mind_ir::GraphProto &graph_proto, LiteGraph::SubGraph *sub_graph, in ConvertTensors()
192 …irModelLoader::ConvertNodes(const mind_ir::GraphProto &graph_proto, LiteGraph::SubGraph *sub_graph, in ConvertNodes()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc56 int ModifySubGraphInputCNodeFormat(const FuncGraphPtr &sub_graph, const ParameterPtr &certain_input, in ModifySubGraphInputCNodeFormat()
98 int ModifyWhileBodyGraphInputs(const CNodePtr &cnode, const FuncGraphPtr &sub_graph, const Paramete… in ModifyWhileBodyGraphInputs()
326 STATUS InferShapePass::SetSubGraphInput(const CNodePtr &cnode, const FuncGraphPtr &sub_graph) { in SetSubGraphInput()
391 STATUS InferShapePass::SetSubGraphOutput(const FuncGraphPtr &sub_graph) { in SetSubGraphOutput()
423 STATUS InferShapePass::SetSubGraphAbstract(const CNodePtr &cnode, const FuncGraphPtr &sub_graph) { in SetSubGraphAbstract()
478 auto &sub_graph = iter.first; in ResetSubGraphInput() local
Dcore_infershape_pass.cc53 int ModifyWhileBodyGraphInputs(const CNodePtr &cnode, const FuncGraphPtr &sub_graph, const Paramete… in ModifyWhileBodyGraphInputs()
253 STATUS CoreInferShapePass::SetSubGraphInput(const CNodePtr &cnode, const FuncGraphPtr &sub_graph) { in SetSubGraphInput()
310 STATUS CoreInferShapePass::SetSubGraphOutput(const FuncGraphPtr &sub_graph) { in SetSubGraphOutput()
340 STATUS CoreInferShapePass::SetSubGraphAbstract(const CNodePtr &cnode, const FuncGraphPtr &sub_graph in SetSubGraphAbstract()
384 auto &sub_graph = iter.first; in ResetSubGraphInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dadd_stitch_atomic_clean_gpu.cc71 CNodePtr StitchAtomicCleanInserter::CreateAssignNode(const FuncGraphPtr &sub_graph, const AnfNodePt… in CreateAssignNode()
86 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(composite_node); in ProcessOriginCNode() local
137 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in FindInnerCNodeUsers() local
156 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in IsStitchWithAtomic() local
Ddeal_with_side_effect.cc27 void DealWithSideEffect::MarkSideEffect(const FuncGraphPtr &sub_graph) { in MarkSideEffect()
56 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
Dinplace_assign_builder.cc37 CNodePtr CreateAssign(const FuncGraphPtr &sub_graph, in CreateAssign()
95 const FuncGraphPtr &sub_graph, in CreateAssignNodeAndCorrectReturn()
181 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(composite_node); in ProcessOriginCNode() local
Dreduce_fake_out_mem.cc76 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
Ddepend_elimination.cc82 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(input1); in Process() local
Dtsa_atomic_add_to_first_tensor.cc61 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(cnode); in FindTsaFirstRealInputInGraph() local
180 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(composite_node); in ProcessOriginalCNode() local
Dcsr_atomic_add.cc58 auto sub_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in FindCandidate() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/
Dcontrol_subgraph_creator.cc26 kernel::SubGraphKernel *sub_graph = nullptr; in CreateControlSubgraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/
Dgraph_converter.cc127const FuncGraphPtr &sub_graph, in BuildTrtNode()
197 FuncGraphPtr sub_graph; in ReplaceSubgraphWithTrtNode() local
263 const Subgraph &sub_graph = graph_partition->CreateNewGraph(segment.second); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dfill_tests.cc83 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_lite_ke… in TEST_F() local
153 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_lite_ke… in TEST_F() local
Dcast_tests.cc117 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_lite_ke… in TEST_F() local
227 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_lite_ke… in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dswitch_not_cut.cc44 auto sub_graph = common::AnfAlgo::GetValueNodeFuncGraph(cnode->input(kIndex1)); in IsValidInlinePartial() local
158 const auto sub_graph = GetValueNode<FuncGraphPtr>(get_item_cnode->input(0)); in IsLazyInlineCall() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Dom_generator.cc85 std::string GetOutNodesStr(const api::FuncGraphManagerPtr &manager, const Subgraph &sub_graph, in GetOutNodesStr()
229 …ator::GenerateMapperConfig(const api::FuncGraphPtr &func_graph, const Subgraph &sub_graph, int cus… in GenerateMapperConfig()
352 int OmGenerator::Run(const api::FuncGraphPtr &func_graph, const Subgraph &sub_graph, int custom_id, in Run()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_capture/
Dspecial_func_infer.cc85 static bool CallNodeReturnConst(CallNode *call_node, Graph *sub_graph, AObject *value) { in CallNodeReturnConst()
98 bool GuardConstCallNodeParam(CallNode *call_node, Graph *sub_graph, int max_guard_depth) { in GuardConstCallNodeParam()
212 Graph *sub_graph = call_node->GetSubGraph(); in InferPrimitive() local
554 Graph *sub_graph = call_node->GetSubGraph(); in InferBuiltinFuncOrMethod() local
582 Graph *sub_graph = call_node->GetSubGraph(); in InferTensorAsType() local
657 Graph *sub_graph = builder->GetGraph(); in InferListAppend() local
716 Graph *sub_graph = builder->GetGraph(); in InferDictPop() local
744 Graph *sub_graph = call_node->GetSubGraph(); in InferMsApiFunc() local
Dbytecode_inliner.cc220 static bool EliminateSideEffect(Graph *top_graph, Graph *sub_graph) { in EliminateSideEffect()
244 static bool CanIninePartial(Graph *top_graph, Graph *sub_graph) { in CanIninePartial()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/
Dunused_node_remove_pass.cc52 auto sub_graph = GetValueNode<FuncGraphPtr>(node); in ProcessGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_graph_executor.h41 std::shared_ptr<TensorRTSubGraph> sub_graph = nullptr; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dkernel_graph_mgr.h46 AnfNodePtr sub_graph; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dgraph_kernel_pass_manager_lite.cc272 void DumpShape(const AnfNodePtr &nd, const FuncGraphPtr &sub_graph, const std::shared_ptr<SubGraphI… in DumpShape()
339 void DumpCNode(const CNodePtr &nd, const FuncGraphPtr &sub_graph, OrderedMap<AnfNodePtr, int32_t> *… in DumpCNode()
378 FuncGraphPtr sub_graph = nd->func_graph(); in DumpIRInSubgraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Druntime_pass.cc261 auto sub_graph = reinterpret_cast<kernel::SubGraphKernel *>(pre_kernel); in DeleteRedundantTrans() local
352 auto sub_graph = reinterpret_cast<kernel::SubGraphKernel *>(subgraph); in GraphOptimizePass() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/const_fold/
Dfold_with_infershape.cc68 auto sub_graph = GetValueNode<FuncGraphPtr>(cnode->input(i)); in HandleCommonFold() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_runtime/
Ddefault_graph_runtime.cc219 auto sub_graph = dynamic_cast<kernel::SubGraphKernel *>(execution_plan_->GetKernels()[0]); in SelectExecutor() local

123