Home
last modified time | relevance | path

Searched refs:set_algo_parameters (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/tests/ut/python/parallel/
Dtest_auto_parallel_star_partial_strategy.py24 from mindspore.parallel import set_algo_parameters
83 set_algo_parameters(fully_use_devices=False)
98 set_algo_parameters(fully_use_devices=False)
113 set_algo_parameters(fully_use_devices=False)
128 set_algo_parameters(fully_use_devices=False)
Dtest_cus_matmul_dds.py25 from mindspore.parallel import set_algo_parameters
138 set_algo_parameters(fully_use_devices=False)
163 set_algo_parameters(fully_use_devices=False)
172 set_algo_parameters(fully_use_devices=False)
181 set_algo_parameters(fully_use_devices=False)
Dtest_set_auto_parallel_context.py18 from mindspore.parallel import set_algo_parameters
82 set_algo_parameters(tensor_slice_align_size=0)
85 set_algo_parameters(tensor_slice_align_size=1025)
Dtest_auto_parallel_resnet_predict.py18 from mindspore.parallel import set_algo_parameters
33 set_algo_parameters(elementwise_op_strategy_follow=True)
Dtest_auto_parallel_two_bn.py24 from mindspore.parallel import set_algo_parameters
77 set_algo_parameters(elementwise_op_strategy_follow=True)
Dtest_auto_parallel_parameter_cast.py25 from mindspore.parallel import set_algo_parameters
62 set_algo_parameters(elementwise_op_strategy_follow=True)
Dtest_dsd_matmul.py25 from mindspore.parallel import set_algo_parameters
154 set_algo_parameters(fully_use_devices=False)
179 set_algo_parameters(fully_use_devices=False)
Dtest_auto_parallel_resnet.py30 from mindspore.parallel import set_algo_parameters
292 set_algo_parameters(elementwise_op_strategy_follow=True)
668 set_algo_parameters(elementwise_op_strategy_follow=True)
695 set_algo_parameters(elementwise_op_strategy_follow=True)
722 set_algo_parameters(enable_algo_approxi=True)
748 set_algo_parameters(elementwise_op_strategy_follow=True)
Dtest_auto_parallel_two_matmul.py27 from mindspore.parallel import set_algo_parameters, get_algo_parameters, reset_algo_parameters
115 set_algo_parameters(tensor_slice_align_enable=False, tensor_slice_align_size=32,
Dtest_parallel_transformer.py32 from mindspore.parallel import set_algo_parameters
462 set_algo_parameters(fully_use_devices=False)
481 set_algo_parameters(fully_use_devices=False)
500 set_algo_parameters(fully_use_devices=False)
519 set_algo_parameters(fully_use_devices=False)
Dtest_reshape.py29 from mindspore.parallel import set_algo_parameters
593 set_algo_parameters(fully_use_devices=False)
635 set_algo_parameters(fully_use_devices=False)
664 set_algo_parameters(fully_use_devices=False)
Dtest_auto_parallel_resnet_sharding_propagation2.py29 from mindspore.parallel import set_algo_parameters
293 set_algo_parameters(elementwise_op_strategy_follow=True)
Dtest_auto_parallel_resnet_sharding_propagation.py30 from mindspore.parallel import set_algo_parameters
295 set_algo_parameters(elementwise_op_strategy_follow=True)
/third_party/mindspore/mindspore/parallel/
D__init__.py17 set_algo_parameters
Dalgo_parameter_config.py223 def set_algo_parameters(**kwargs): function
/third_party/mindspore/tests/st/auto_parallel/
Doptimizer_parallel.py36 from mindspore.parallel import set_algo_parameters
278 set_algo_parameters(fully_use_devices=False)
291 set_algo_parameters(fully_use_devices=False)
Dresnet50_expand_loss.py29 from mindspore.parallel import set_algo_parameters
363 set_algo_parameters(elementwise_op_strategy_follow=True)
Dparallel_strategy_search.py35 from mindspore.parallel import set_algo_parameters
301 set_algo_parameters(fully_use_devices=False)