Home
last modified time | relevance | path

Searched refs:computation_time_parameter (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/tests/ut/python/parallel/
Dtest_allreduce_fusion.py177 computation_time_parameter = cost_model_context.get_cost_model_context(
179 assert computation_time_parameter == 0.2
181 computation_time_parameter = cost_model_context.get_cost_model_context(
183 assert computation_time_parameter == 0.1
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcostmodel_context.cc206 …ext::set_costmodel_allreduce_fusion_computation_time_parameter(double computation_time_parameter) { in set_costmodel_allreduce_fusion_computation_time_parameter() argument
207 costmodel_allreduce_fusion_computation_time_parameter_ = computation_time_parameter; in set_costmodel_allreduce_fusion_computation_time_parameter()
/third_party/mindspore/mindspore/parallel/
D_cost_model_context.py450 def set_costmodel_allreduce_fusion_computation_time_parameter(self, computation_time_parameter): argument
462 …ontext_handle.set_costmodel_allreduce_fusion_computation_time_parameter(computation_time_parameter)