| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_fillv2.py | 40 def compile_graph(net, device_num, parallel_mode, y, search_mode="recursive_programming"): function
|
| D | test_gpu_dropout.py | 65 def compile_graph(net, device_num, parallel_mode, x, y, search_mode="dynamic_programming"): function
|
| D | test_dsd_matmul.py | 123 def compile_graph(batch_size, num_heads, dp, mp, auto=False, shard=True): function
|
| D | test_unsortedsegmentmin.py | 70 def compile_graph(x, y, segments, strategy1, strategy2, auto=False): function
|
| D | test_cus_matmul_dds.py | 107 def compile_graph(batch_size, num_heads, dp, mp, auto=False, shard=True): function
|
| D | test_unsortedsegmentprod.py | 67 def compile_graph(x, y, segments, strategy1, strategy2, auto=False): function
|
| D | test_unsortedsegmentmax.py | 70 def compile_graph(x, y, segments, strategy1, strategy2, auto=False): function
|
| D | test_unsortedsegmentsum.py | 69 def compile_graph(x, y, segments, strategy1, strategy2, auto=False): function
|
| D | test_gather_batch_dims.py | 75 def compile_graph(net, device_num, parallel_mode, x, y): function
|
| D | test_onehot.py | 72 def compile_graph(strategy1, strategy2, strategy3, strategy4, auto=False, onthot_axis=-1): function
|
| D | test_index_select.py | 91 def compile_graph(net, device_num, parallel_mode, x, y, search_mode="dynamic_programming"): function
|
| D | test_softmax_gather_net.py | 73 def compile_graph(net, device_num, parallel_mode, x, y, search_mode="dynamic_programming"): function
|
| D | test_virtual_output.py | 128 def compile_graph(x, net): function
|
| D | test_gather_v2.py | 87 def compile_graph(net, device_num, parallel_mode, x, y, search_mode="dynamic_programming"): function
|
| D | test_reshape_shard_propagation.py | 77 def compile_graph(net, device_num, x): function
|
| D | test_auto_parallel_reshape.py | 86 def compile_graph(net, parallel_mode, device_num, x, search_mode="dynamic_programming"): function
|