Home
last modified time | relevance | path

Searched refs:set_multi_subgraphs (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/parallel/
D_cost_model_context.py217 def set_multi_subgraphs(self, multi_subgraph): member in _CostModelContext
229 self._context_handle.set_multi_subgraphs(multi_subgraph)
629 cost_model_context().set_multi_subgraphs(multi_subgraph)
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcostmodel_context.h95 void set_multi_subgraphs(bool);
Dcostmodel_context.cc174 void CostModelContext::set_multi_subgraphs(bool multi_graphs) { in set_multi_subgraphs() function in mindspore::parallel::CostModelContext
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc235 ….def("set_multi_subgraphs", &CostModelContext::set_multi_subgraphs, "Set the parameter is_multi_su… in PYBIND11_MODULE()