Wanchao Liang 3f89b779c4 [jit] allow submodule methods inference rule be different (#43872) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43872 This PR allows the recursive scripting to have a separate submodule_stubs_fn to create its submodule with specific user provided rules. Fixes https://github.com/pytorch/pytorch/issues/43729 Test Plan: Imported from OSS Reviewed By: suo Differential Revision: D23430176 Pulled By: wanchaol fbshipit-source-id: 20530d7891ac3345b36f1ed813dc9c650b28d27a		2020-09-23 14:10:31 -07:00
..
mobile	Move async + serialization implementation out of 'jit/__init__.py' (#41018)	2020-07-09 10:10:01 -07:00
__init__.py	[JIT] Add JIT support for torch.no_grad (#41371)	2020-08-27 15:32:57 -07:00
_async.py	move misc implementation out of jit/__init__.py (#41154)	2020-07-13 16:59:55 -07:00
_builtins.py	[JIT] Add mypy type annotations for JIT (#43862)	2020-09-03 15:09:24 -07:00
_freeze.py	move misc implementation out of jit/__init__.py (#41154)	2020-07-13 16:59:55 -07:00
_fuser.py	Respect canFuseOn{CPU,GPU} in TE fuser (#43967)	2020-09-02 18:00:25 -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	[jit] allow submodule methods inference rule be different (#43872)	2020-09-23 14:10:31 -07:00
_script.py	[jit] allow submodule methods inference rule be different (#43872)	2020-09-23 14:10:31 -07:00
_serialization.py	add typing annotations for a few torch.utils.* modules (#43806)	2020-09-11 10:20:55 -07:00
_state.py	[JIT] Add mypy type annotations for JIT (#43862)	2020-09-03 15:09:24 -07:00
_trace.py	[jit] allow submodule methods inference rule be different (#43872)	2020-09-23 14:10:31 -07:00
annotations.py	Remove EXPERIMENTAL_ENUM_SUPPORT feature guard (#44243)	2020-09-16 11:45:59 -07:00
frontend.py	[JIT] Add default arguments for class types (#45098)	2020-09-22 18:37:44 -07:00
quantized.py	Enable typechecks for torch.nn.quantized.modules.linear (#44154)	2020-09-03 19:52:49 -07:00
supported_ops.py	[JIT] Add mypy type annotations for JIT (#43862)	2020-09-03 15:09:24 -07:00
unsupported_tensor_ops.py	[JIT] Add mypy type annotations for JIT (#43862)	2020-09-03 15:09:24 -07:00