pytorch
7ac872b9 - [JIT] Modify to_backend API so that it accepts wrapped modules (#43612)

Commit
4 years ago
[JIT] Modify to_backend API so that it accepts wrapped modules (#43612) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43612 **Summary** This commit modifies the `torch._C._jit_to_backend` function so that it accepts `ScriptModules` as inputs. It already returns `ScriptModules` (as opposed to C++ modules), so this makes sense and makes the API more intuitive. **Test Plan** Continuous integration, which includes unit tests and out-of-tree tests for custom backends. **Fixes** This commit fixes #41432. Test Plan: Imported from OSS Reviewed By: suo, jamesr66a Differential Revision: D23339854 Pulled By: SplitInfinity fbshipit-source-id: 08ecef729c4e1e6bddf3f483276947fc3559ea88
Author
Meghan Lele
Parents
Loading