mindspore2022/mindspore/_extends/graph_kernel/model
hanhuifeng2020 bc46d644fe [GraphKernel]Support reshape/elewise/broadcast+transdata fusion 2021-05-08 11:42:14 +08:00
..
__init__.py support parallel fusion 2021-01-27 17:27:13 +08:00
graph_parallel.py pipeline optimization for parallel fusion 2021-03-09 17:07:24 +08:00
graph_split.py [GraphKernel]Support reshape/elewise/broadcast+transdata fusion 2021-05-08 11:42:14 +08:00
model.py reduce recursion overhead of split model 2021-04-25 21:27:09 +08:00
model_builder.py [GraphKernel]Support reshape/elewise/broadcast+transdata fusion 2021-05-08 11:42:14 +08:00
op_infer.py adapt expanders of some ops from gpu to ascend 2021-05-04 16:32:45 +08:00