pytorch
2a538971 - [jit][tensorexpr] Added aten::batch_norm into fuser when in inference mode (#54204)

Commit
3 years ago
[jit][tensorexpr] Added aten::batch_norm into fuser when in inference mode (#54204) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54204 Test Plan: Imported from OSS Reviewed By: bertmaher Differential Revision: D27134348 Pulled By: huiguoo fbshipit-source-id: 5ea7a6c5bc694fcdfc436dba3fa6eb269420324e
Author
Parents
Loading