Searched defs:compile_graph (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_dsd_matmul.py | 118 def compile_graph(batch_size, num_heads, dp, mp, auto=False, shard=True): function
|
| D | test_cus_matmul_dds.py | 102 def compile_graph(batch_size, num_heads, dp, mp, auto=False, shard=True): function
|
| D | test_unsortedsegmentmax.py | 66 def compile_graph(x, y, segments, strategy1, strategy2, auto=False): function
|
| D | test_unsortedsegmentmin.py | 66 def compile_graph(x, y, segments, strategy1, strategy2, auto=False): function
|
| D | test_unsortedsegmentsum.py | 65 def compile_graph(x, y, segments, strategy1, strategy2, auto=False): function
|
| D | test_onehot.py | 68 def compile_graph(strategy1, strategy2, strategy3, strategy4, auto=False, onthot_axis=-1): function
|
| D | test_virtual_output.py | 118 def compile_graph(x, net): function
|