Searched refs:set_costmodel_allreduce_fusion_algorithm (Results 1 – 4 of 4) sorted by relevance
298 def set_costmodel_allreduce_fusion_algorithm(self, algorithm): member in _CostModelContext310 self._context_handle.set_costmodel_allreduce_fusion_algorithm(algorithm)512 …stmodel_allreduce_fusion_algorithm": cost_model_context().set_costmodel_allreduce_fusion_algorithm,
98 void set_costmodel_allreduce_fusion_algorithm(int64_t);
182 void CostModelContext::set_costmodel_allreduce_fusion_algorithm(int64_t algorithm) { in set_costmodel_allreduce_fusion_algorithm() function in mindspore::parallel::CostModelContext
239 …costmodel_allreduce_fusion_algorithm", &CostModelContext::set_costmodel_allreduce_fusion_algorithm, in PYBIND11_MODULE()