Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/
D_cost_model_context.py303 def set_costmodel_allreduce_fusion_algorithm(self, algorithm): member in _CostModelContext
315 self._context_handle.set_costmodel_allreduce_fusion_algorithm(algorithm)
542 …stmodel_allreduce_fusion_algorithm": cost_model_context().set_costmodel_allreduce_fusion_algorithm,
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dcostmodel_context.h98 void set_costmodel_allreduce_fusion_algorithm(int64_t algorithm);
Dcostmodel_context.cc191 void CostModelContext::set_costmodel_allreduce_fusion_algorithm(int64_t algorithm) { in set_costmodel_allreduce_fusion_algorithm() function in mindspore::parallel::CostModelContext
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dinit.cc458 …costmodel_allreduce_fusion_algorithm", &CostModelContext::set_costmodel_allreduce_fusion_algorithm, in PYBIND11_MODULE()