pytorch
0b3755b1 - Add optimization blacklist as second arg to optimizeForMobile method. (#37462)

Commit
4 years ago
Add optimization blacklist as second arg to optimizeForMobile method. (#37462) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/37462 Instead of running all the optimization pass in optimizeForMobile method, introducing a whitelist optimizer dictionary as second param in the method, when it is not passed during calling, the method will run all the optimization passes, otherwise the method will read the dict and only run the pass with value of True. ghstack-source-id: 106104503 Test Plan: python test/test_mobile_optimizer.py Imported from OSS Differential Revision: D22096029 fbshipit-source-id: daa9370c0510930f4c032328b225df0bcf97880f
Author
Parents
Loading