Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/parallel/
D_cost_model_context.py425 def set_costmodel_allreduce_fusion_allreduce_bandwidth(self, allreduce_bandwidth): member in _CostModelContext
437 self._context_handle.set_costmodel_allreduce_fusion_allreduce_bandwidth(allreduce_bandwidth)
519 cost_model_context().set_costmodel_allreduce_fusion_allreduce_bandwidth,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcostmodel_context.h115 void set_costmodel_allreduce_fusion_allreduce_bandwidth(double);
Dcostmodel_context.cc202 void CostModelContext::set_costmodel_allreduce_fusion_allreduce_bandwidth(double allreduce_bandwidt… in set_costmodel_allreduce_fusion_allreduce_bandwidth() function in mindspore::parallel::CostModelContext
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc262 &CostModelContext::set_costmodel_allreduce_fusion_allreduce_bandwidth, in PYBIND11_MODULE()