Searched refs:cut_list (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/acl_vm/ |
| D | acl_vm.h | 41 const std::vector<PrimitivePtr> &cut_list);
58 const std::vector<PrimitivePtr> &cut_list);
|
| D | acl_vm.cc | 113 const std::vector<PrimitivePtr> &cut_list) in AclCompileGraph() argument
114 : CompileGraph(backend, cut_list) {} in AclCompileGraph()
288 const std::vector<PrimitivePtr> &cut_list) in AclCompileGraphs() argument
289 : CompileGraphs(backend, cut_list) { in AclCompileGraphs()
292 transform_ = std::make_shared<AclCompileGraph>(backend, cut_list); in AclCompileGraphs()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | transform.h | 54 …explicit CompileGraph(const BackendPtr &backend, const std::vector<PrimitivePtr> &cut_list = nonli…
115 …explicit CompileGraphs(const BackendPtr &backend, const std::vector<PrimitivePtr> &cut_list = nonl…
|
| D | graph_partition.h | 35 …explicit GraphPartition(const std::vector<PrimitivePtr> &cut_list, const std::string &backend_name…
|
| D | transform.cc | 60 CompileGraph::CompileGraph(const BackendPtr &backend, const std::vector<PrimitivePtr> &cut_list) : … in CompileGraph() argument
66 graph_partition_ = std::make_shared<GraphPartition>(cut_list, backend->name()); in CompileGraph()
498 CompileGraphs::CompileGraphs(const BackendPtr &backend, const std::vector<PrimitivePtr> &cut_list) … in CompileGraphs() argument
501 transform_ = std::make_shared<CompileGraph>(backend, cut_list); in CompileGraphs()
|
| D | graph_partition.cc | 621 GraphPartition::GraphPartition(const std::vector<PrimitivePtr> &cut_list, const std::string &backen… in GraphPartition() argument
622 : cut_list_(cut_list), backend_name_(backend_name) {} in GraphPartition()
|
| D | backend.cc | 353 auto &cut_list = pynative_mode ? compile::control_ops : GetMsNonlinearOps(); in MindRTBackend() local
355 graph_partition_ = std::make_shared<GraphPartition>(cut_list, backend_name); in MindRTBackend()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | action.cc | 743 std::vector<PrimitivePtr> cut_list = compile::nonlinear_ops; in TaskEmitAction() local
745 cut_list = compile::GetMsNonlinearOps(); in TaskEmitAction()
747 std::shared_ptr<CompileGraphs> compile = std::make_shared<CompileGraphs>(bc_ptr, cut_list); in TaskEmitAction()
|