Home
last modified time | relevance | path

Searched defs:sub_graphs (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dsub_graph_split.cc129 std::vector<bool> *cor_group, std::vector<Subgraph> *sub_graphs) { in dfs()
222 void SearchSubGraph::ConvertSubGraphToModel(std::vector<Subgraph> *sub_graphs) { in ConvertSubGraphToModel()
441 void SearchSubGraph::OptimizeAfterFusion(std::vector<Subgraph> *sub_graphs, uint32_t root_node_inde… in OptimizeAfterFusion()
689 void SearchSubGraph::InitSubgraphRuntimeInfo(std::vector<Subgraph> *sub_graphs) { in InitSubgraphRuntimeInfo()
734 void SearchSubGraph::SubgraphFusion(std::vector<Subgraph> *sub_graphs) { in SubgraphFusion()
775 void SearchSubGraph::CalculateCostModel(std::vector<Subgraph> *sub_graphs) { in CalculateCostModel()
Dlite_model.h285 auto sub_graphs = meta_graph.subGraph(); in GenerateModel() local
Druntime_pass.cc349 STATUS GraphOptimizePass(std::vector<kernel::KernelExec *> *sub_graphs) { in GraphOptimizePass()
Dlite_session.cc775 const auto &sub_graphs = model->graph_.sub_graphs_; in SetInitRefCountOfPartialSubgraphInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dgraph_kernel_pass_manager_lite.cc370 … OrderedMap<FuncGraphPtr, std::shared_ptr<SubGraphIRInfo>> *const sub_graphs, int32_t total_para, in DumpIRInSubgraph()
407 void DumpSubgraph(const OrderedMap<FuncGraphPtr, std::shared_ptr<SubGraphIRInfo>> *sub_graphs, in DumpSubgraph()
500 OrderedMap<FuncGraphPtr, std::shared_ptr<SubGraphIRInfo>> sub_graphs; in DumpIR() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/debug/
Ddraw.cc46 …std::vector<AnfNodePtr> &nodes, OrderedMap<FuncGraphPtr, std::shared_ptr<BaseDigraph>> *sub_graphs, in DrawNodes()
72 OrderedMap<FuncGraphPtr, std::shared_ptr<BaseDigraph>> *sub_graphs) { in DrawValueNodes()
138 OrderedMap<FuncGraphPtr, std::shared_ptr<BaseDigraph>> sub_graphs; in DrawByOpt() local
Danf_ir_dump.cc1069 … OrderedMap<FuncGraphPtr, std::shared_ptr<SubGraphIRInfo>> *const sub_graphs, int32_t total_para, in DumpIRInSubgraph()
1110 void DumpSubgraph(const OrderedMap<FuncGraphPtr, std::shared_ptr<SubGraphIRInfo>> *sub_graphs, in DumpSubgraph()
1296 void DumpNodeCounting(const OrderedMap<FuncGraphPtr, std::shared_ptr<SubGraphIRInfo>> *sub_graphs, in DumpNodeCounting()
1447 auto sub_graphs = top_func->func_graphs_used_total(); in DumpIRHead() local
1482 OrderedMap<FuncGraphPtr, std::shared_ptr<SubGraphIRInfo>> sub_graphs; in DumpIR() local
1530 OrderedMap<FuncGraphPtr, std::shared_ptr<SubGraphIRInfo>> sub_graphs; in DumpIRForRDR() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_device_context.cc93 const auto &sub_graphs = mng->func_graphs(); in PartitionGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dbackend_base.cc783 const auto &sub_graphs = root_graph->manager()->func_graphs_used_total(root_graph); in UnifyMindIR() local
801 const auto &sub_graphs = manager->func_graphs_used_total(root_graph); in CompileSubGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/debug/
Dtrace.cc592 auto sub_graphs = top_func->func_graphs_used_total(); in ExportFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tflite/
Dtflite_model_parser.cc671 auto sub_graphs = node_vs_graph.second; in ControlFlowNodePostProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dcontrol_node_parser.cc1577 FuncGraphSet sub_graphs = root_func_graph_->manager()->func_graphs(); in ParseDeviceContextForFuncGraph() local