Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/parallel/
D_auto_parallel_context.py202 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/
Dparallel_strategy_search.py299 …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/
Dcontext.py346auto_parallel_search_mode=str, parameter_broadcast=bool, strategy_ckpt_load_file=str,