pytorch

70d6446a - 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