Wonjoo Lee a171b0636a Add use_lazy_shape flag to GenLazyIr class (#88444) ... Add use_lazy_shape flag to GenLazyIr class to allow XLA to use its custom shape class. The default value is kept to use lazy shape, so this PR does not introduce any new behaviors. PyTorch/XLA companion PR: https://github.com/pytorch/xla/pull/4111 Pull Request resolved: https://github.com/pytorch/pytorch/pull/88444 Approved by: https://github.com/alanwaketan, https://github.com/wconstab		2022-11-04 08:23:56 +00:00
..
__init__.py	Apply ufmt linter to all py files under torchgen (#81570)	2022-07-16 03:52:25 +00:00
lazy_ir.py	Add use_lazy_shape flag to GenLazyIr class (#88444)	2022-11-04 08:23:56 +00:00
lazy_ts_lowering.py	Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)	2022-07-26 01:20:44 +00:00
native_functions.py	Apply ufmt linter to all py files under torchgen (#81570)	2022-07-16 03:52:25 +00:00
register_dispatch_key.py	Revert "Revert "Symintifying slice ops (#85196)"" (#85746)	2022-09-28 04:37:35 +00:00
ufunc.py	Use OrderedSet in ufunc codegen (#82567)	2022-08-01 18:11:06 +00:00