Searched defs:subgraphs (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/graph_kernel/ |
| D | splitter.py | 117 def _dump_split_info(use_repo, graph_str, graph, subgraphs, graph_mode, graph_list): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/ |
| D | graph_split_api.cc | 180 std::vector<Subgraph> subgraphs; in GenerateSubgraphs() local
298 … auto subgraphs = GenerateSubgraphs(func_graph, graph_total_cnodes, segment_infos, &subgraph_cnt); in GraphSplit() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mock/ |
| D | transform_mock.cc | 60 …erImpl::SetOpSubgraphFunc(const OperatorPtr &op, const std::shared_ptr<std::vector<DfGraph>> &subg… in SetOpSubgraphFunc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | sub_graph_split.cc | 83 bool SearchSubGraph::CheckIsParallelSubGraph(const std::vector<Subgraph> &subgraphs) { in CheckIsParallelSubGraph()
820 std::vector<Subgraph> &subgraphs = map.second; in SubGraphSplitByMiddle() local
882 std::vector<Subgraph> &subgraphs = map.second; in SubGraphSplitByOffLineParallel() local
|
| D | runtime_pass.cc | 327 STATUS RuntimePass(std::vector<kernel::KernelExec *> *subgraphs, std::vector<Tensor *> *tensors) { in RuntimePass()
|
| D | kernel_exec_util.cc | 491 SubGraphKernel *KernelExecUtil::BelongToWhichSubGraph(const std::vector<KernelExec *> &subgraphs, K… in BelongToWhichSubGraph()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/ |
| D | control_flow_scheduler.cc | 79 auto subgraphs = partial_kernel->subgraph_kernels(); in SplitNonTailCallSubGraphs() local
533 auto subgraphs = partial_kernel->subgraph_kernels(); in GetTailCallFinalSubgraphs() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/stub/transform/ |
| D | util.cc | 105 …erImpl::SetOpSubgraphFunc(const OperatorPtr &op, const std::shared_ptr<std::vector<DfGraph>> &subg… in SetOpSubgraphFunc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter.h | 318 Status SetOpSubgraphFunc(const OperatorPtr &op, std::shared_ptr<std::vector<DfGraph>> subgraphs) { in SetOpSubgraphFunc()
322 …void setSubgraph(const OperatorPtr &op, std::shared_ptr<std::vector<DfGraph>> subgraphs) override { in setSubgraph()
|
| D | op_adapter.cc | 300 …erImpl::SetOpSubgraphFunc(const OperatorPtr &op, const std::shared_ptr<std::vector<DfGraph>> &subg… in SetOpSubgraphFunc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | stage_compute.cc | 831 std::vector<FuncGraphPtr> subgraphs; in ChangeStageNumber() local
|