pytorch/torch/jit
Jerry Zhang ea273c68f9
Inplace construct of TorchScript Module and inplace option for quantization (#40750)
* [WIP][JIT] Add ScriptModule._reconstruct (#39979)

Summary:
**Summary**
This commit adds an instance method `_reconstruct` that permits users
to reconstruct a `ScriptModule` from a given C++ `Module` instance.

**Testing**
This commit adds a unit test for `_reconstruct`.

**Fixes**
This pull request fixes https://github.com/pytorch/pytorch/issues/33912.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/39979

Differential Revision: D22172323

Pulled By: SplitInfinity

fbshipit-source-id: 9aa6551c422a5a324b822a09cd8d7c660f99ca5c

* [quant][graphmode] Enable inplace option for top level API (#40414)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40414

after `_reconstruct` is supported in RecursiveScriptModule: https://github.com/pytorch/pytorch/pull/39979
we can support inplace option in quantization API

Test Plan: Imported from OSS

Differential Revision: D22178326

fbshipit-source-id: c78bc2bcf2c42b06280c12262bb31aebcadc6c32

Co-authored-by: Meghan Lele <meghanl@fb.com>
2020-07-02 10:22:45 -07:00
..
mobile [fbcode] Add find_method into lite interpreter python binding. 2020-06-13 07:48:13 -07:00
__init__.py Inplace construct of TorchScript Module and inplace option for quantization (#40750) 2020-07-02 10:22:45 -07:00
_builtins.py [jit] register distributed backward (#38494) 2020-06-08 19:43:40 -07:00
_logging.py Experimental logging/counters API (#18235) 2019-03-29 17:14:03 -07:00
_pickle.py [jit] Add type tags to lists/dicts in pickle (#33255) 2020-03-10 19:17:01 -07:00
_recursive.py [distribtued.nn] Implement TorchScript-compatible RemoteModule API (#37139) 2020-06-15 19:07:35 -07:00
annotations.py [JIT x RPC] Consolidate RRef type class and RRef impl class (#35694) 2020-06-18 07:46:38 -07:00
frontend.py [JIT] Add support for with statements (#34705) 2020-06-18 16:57:18 -07:00
quantized.py Dynamic quantization support for LSTMCell, RNNCell and GRUCell [Remove randomness in weights] (#40102) 2020-06-16 21:29:50 -07:00
supported_ops.py DOC: fix import error (#38921) 2020-05-26 15:58:34 -07:00
unsupported_tensor_ops.py Revert "Revert D21337640: [pytorch][PR] Split up documentation into subpages and clean up some warnings" (#37778) 2020-05-04 14:32:35 -07:00