| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_hybird_parallel_activation.py | 58 def __init__(self, strategy1, strategy2, strategy3): argument
84 def __init__(self, strategy1, strategy2, strategy3): argument
110 def __init__(self, strategy1, strategy2, strategy3): argument
136 def __init__(self, strategy1, strategy2, strategy3): argument
162 def __init__(self, strategy1, strategy2, strategy3): argument
191 def __init__(self, strategy1, strategy2, strategy3, strategy4, strategy5, strategy6): argument
223 def __init__(self, strategy1, strategy2, strategy3, strategy4, strategy5, strategy6): argument
|
| D | test_reduce_method_info.py | 85 def __init__(self, strategy1, strategy2, strategy3): argument
112 def __init__(self, strategy1, strategy2, strategy3): argument
139 def __init__(self, strategy1, strategy2, strategy3): argument
166 def __init__(self, strategy1, strategy2, strategy3): argument
262 def __init__(self, strategy1, strategy2, strategy3): argument
289 def __init__(self, strategy1, strategy2, strategy3): argument
316 def __init__(self, strategy1, strategy2, strategy3): argument
343 def __init__(self, strategy1, strategy2, strategy3): argument
357 def __init__(self, strategy1, strategy2, strategy3): argument
444 def __init__(self, strategy1, strategy2, strategy3): argument
[all …]
|
| D | test_layer_norm_further.py | 28 strategy1=None, strategy2=None, strategy3=None): argument
52 strategy1=None, strategy2=None, strategy3=None): argument
76 strategy1=None, strategy2=None, strategy3=None): argument
100 strategy1=None, strategy2=None, strategy3=None): argument
122 strategy1=None, strategy2=None, strategy3=None): argument
|
| D | test_gathernd_further.py | 48 def __init__(self, w1_shape, indices_shape, strategy1=None, strategy2=None, strategy3=None): argument
64 def __init__(self, w1_shape, indices_shape, strategy1=None, strategy2=None, strategy3=None): argument
79 def __init__(self, w1_shape, indices_shape, strategy1=None, strategy2=None, strategy3=None): argument
|
| D | test_mix_precision_hybrid_parallel.py | 51 def __init__(self, strategy1, strategy2, strategy3): argument
66 def __init__(self, strategy1, strategy2, strategy3, strategy4): argument
|
| D | test_onehot.py | 33 def __init__(self, network, strategy3, strategy4, axis): argument
68 def compile_graph(strategy1, strategy2, strategy3, strategy4, auto=False, onthot_axis=-1): argument
|
| D | test_virtual_dataset_with_strategy.py | 49 def __init__(self, strategy1, strategy2, strategy3): argument
63 def __init__(self, strategy1, strategy2, strategy3): argument
|
| D | test_auto_parallel_shard_propagation2.py | 25 def __init__(self, mul_weight, strategy1=None, strategy2=None, strategy3=None): argument
|
| D | test_auto_parallel_shard_propagation3.py | 25 def __init__(self, mul_weight, strategy1=None, strategy2=None, strategy3=None): argument
|
| D | test_matmul_tensor.py | 61 def __init__(self, strategy1, strategy2, strategy3): argument
115 def __init__(self, strategy1, strategy2, strategy3): argument
|
| D | test_auto_parallel_shard_propagation.py | 25 def __init__(self, mul_weight, strategy1=None, strategy2=None, strategy3=None): argument
|
| D | test_strategy_checkpoint.py | 52 def __init__(self, strategy1, strategy2, strategy3, strategy4, strategy5, strategy6): argument
116 def __init__(self, strategy1, strategy3, strategy4, strategy5, strategy6, strategy7): argument
237 def __init__(self, strategy1, strategy3, strategy4, strategy5): argument
|
| D | test_layer_norm.py | 27 def __init__(self, mul_weight, strategy1=None, strategy2=None, strategy3=None): argument
|
| D | test_one_weight_parameter.py | 30 def __init__(self, network, strategy3): argument
|
| D | test_two_weights_parameter.py | 30 def __init__(self, network, strategy3): argument
|
| D | test_expand_dims.py | 25 def __init__(self, mul_weight, strategy1=None, strategy2=None, strategy3=None): argument
|
| D | test_virtual_dataset_3_input.py | 52 def __init__(self, strategy1, strategy2, strategy3): argument
|
| D | test_l2normalize.py | 53 def __init__(self, strategy1, strategy2, strategy3): argument
|
| D | test_manual_gatherv2.py | 29 strategy3=None, argument
|
| D | test_optimizer_clone_weight.py | 28 def __init__(self, network, strategy3): argument
|
| D | test_self_attention.py | 60 def __init__(self, strategy1, strategy2, strategy3, strategy4, strategy5): argument
|
| D | test_broadcast_to.py | 43 def __init__(self, weight1, strategy1=None, strategy2=None, strategy3=None, is_parameter=True): argument
|
| D | test_manual_embedding_lookup.py | 36 strategy3=None, argument
|
| D | test_range.py | 48 def __init__(self, weight, start, limit, delta, strategy1=None, strategy2=None, strategy3=None): argument
|
| D | test_linear.py | 30 def __init__(self, network, strategy3): argument
|