Searched refs:auto_parallel_search_mode (Results 1 – 3 of 3) sorted by relevance
/third_party/mindspore/mindspore/parallel/ |
D | _auto_parallel_context.py | 202 def set_strategy_search_mode(self, auto_parallel_search_mode): argument 210 ret = self._context_handle.set_strategy_search_mode(auto_parallel_search_mode) 212 … raise ValueError("Strategy search mode does not support {}".format(auto_parallel_search_mode)) 681 loss_repeated_mean=bool, parallel_mode=str, auto_parallel_search_mode=str,
|
/third_party/mindspore/tests/st/auto_parallel/ |
D | parallel_strategy_search.py | 299 …def mindspore_auto_parallel_impl(self, dataset, epoch, device_num, auto_parallel_search_mode="dyna… argument 305 auto_parallel_search_mode=auto_parallel_search_mode) 382 … epoch=2, device_num=8, auto_parallel_search_mode="recursive_programming")
|
/third_party/mindspore/mindspore/ |
D | context.py | 346 … auto_parallel_search_mode=str, parameter_broadcast=bool, strategy_ckpt_load_file=str,
|