mindspore2022/mindspore/_extends/graph_kernel
dayschan 149dab39c5 Add expander for AddN; update akg submodule 2021-06-29 11:43:46 +08:00
..
expanders Add expander for AddN; update akg submodule 2021-06-29 11:43:46 +08:00
model fix recompute_fuse with reshape user 2021-06-24 17:36:35 +08:00
__init__.py support parallel fusion 2021-01-27 17:27:13 +08:00
expander.py [graph kernel] clean code for expanders. 2021-04-26 16:28:37 +08:00
parallel_estimate.py fix static warnings 2021-06-04 17:25:42 +08:00
splitter.py refactor tile op and in expander open on gpu 2021-04-25 17:42:37 +08:00
utils.py dump graph_kernel_split info 2021-03-29 17:15:02 +08:00