pytorch

12658fcd - Support both train / eval modes for ModuleInfo

Support both train / eval modes for ModuleInfo Pull Request resolved: https://github.com/pytorch/pytorch/pull/78735 Approved by: https://github.com/albanD