Boyuan Yao a7cda6f57d [fx] Add offload codegen (#1598) ...

* [fx] add input activation offload to codegen

* [fx] modify unit test

* [fx] remove two skips in torch11

* [fx] use all_input_nodes instead of _input_nodes

2022-09-14 15:49:06 +08:00

6.9 KiB

Raw Blame History

View Raw