Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/parallel/
D_cost_model_context.py323 def set_costmodel_allreduce_fusion_times(self, allreduce_fusion_times): member in _CostModelContext
335 self._context_handle.set_costmodel_allreduce_fusion_times(allreduce_fusion_times)
513 "costmodel_allreduce_fusion_times": cost_model_context().set_costmodel_allreduce_fusion_times,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcostmodel_context.h101 void set_costmodel_allreduce_fusion_times(int64_t);
Dcostmodel_context.cc186 void CostModelContext::set_costmodel_allreduce_fusion_times(int64_t allreduce_fusion_times) { in set_costmodel_allreduce_fusion_times() function in mindspore::parallel::CostModelContext
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc243 …ef("set_costmodel_allreduce_fusion_times", &CostModelContext::set_costmodel_allreduce_fusion_times, in PYBIND11_MODULE()