pytorch
2e49c5dc - Move GetArgumentNamesModule registration to InterpreterManager() (#61549)

Commit
3 years ago
Move GetArgumentNamesModule registration to InterpreterManager() (#61549) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61549 Move GetArgumentNamesModule registration to InterpreterManager() such that the module is a permanent part of the interpreters and can be used by InterpreterSession.global() freely. Test Plan: [... ~/fbsource/fbcode/caffe2] buck test mode/dev caffe2/fb/predictor:pytorch_predictor_test -- PyTorchDeployPredictor.GetArgumentNames Reviewed By: wconstab Differential Revision: D29643460 fbshipit-source-id: cf132d4795cbb334ce164ac715d590a105535508
Author
Jiewen Tan
Parents
Loading