pytorch
e0451199 - [JIT] Add default arguments for class types (#45098)

Commit
4 years ago
[JIT] Add default arguments for class types (#45098) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45098 **Summary** This commit adds support for default arguments in methods of class types. Similar to how default arguments are supported for regular script functions and methods on scripted modules, default values are retrieved from the definition of a TorchScript class in Python as Python objects, converted to IValues, and then attached to the schemas of already compiled class methods. **Test Plan** This commit adds a set of new tests to TestClassType to test default arguments. **Fixes** This commit fixes #42562. Test Plan: Imported from OSS Reviewed By: gmagogsfm Differential Revision: D23844769 Pulled By: SplitInfinity fbshipit-source-id: ceedff7703bf9ede8bd07b3abcb44a0f654936bd
Author
Meghan Lele
Parents
Loading