Searched refs:block_fusion (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/st/graph_kernel/model/ |
| D | test_graph_parallel.py | 36 def block_fusion(graphs): function
42 assert block_fusion([injective_graph([40, 1024]), injective_graph([40, 1024])])
43 assert block_fusion([reduce_graph([1024, 1024], [1]), injective_graph([24, 1024])])
44 assert not block_fusion([reduce_graph([1024, 1024], [1]), injective_graph([50, 1024])])
45 assert not block_fusion([reduce_graph([1024, 1024], [0, 1]), injective_graph([1024, 1024])])
|