Horace He 50cadfae10 Add strictness check and made tensors into leaves if input tensors were leaves (#77474) ... I think this makes sense to do? Otherwise, if you call `backward()` in your traced function, you can't get gradients out of any tensors that should have been leaves. Pull Request resolved: https://github.com/pytorch/pytorch/pull/77474 Approved by: https://github.com/ezyang		2022-05-21 01:16:39 +00:00
..
unification	[lint] upgrade mypy to latest version	2022-05-03 20:51:34 +00:00
__init__.py
accelerator_partitioner.py	[FX] Move graph_manipulation and param_fetch out of experimental and into passes (#65183)	2021-09-17 09:32:40 -07:00
const_fold.py	Add mapping for unsqueeze_n_times (#75043)	2022-04-01 05:02:56 +00:00
debug.py	[torch.fx] add code-gen customizability and support for setting breakpoint in code-gen'd forward() call (#67139)	2021-11-16 13:28:11 -08:00
graph_gradual_typechecker.py	add documentation to shape inference algorithm (#64312)	2021-09-01 18:34:17 -07:00
merge_matmul.py	[fx] fix merge_matmul tests making invalid torch.split calls	2022-05-09 20:04:21 +00:00
meta_tracer.py	[WIP] Make constructor calls in experimental MetaTracer serializable	2022-05-11 00:19:47 +00:00
normalize.py	Preserve codegen on fx graph in transformer (#74189)	2022-03-16 16:33:44 +00:00
optimization.py	Stop writing logs to root logger (#72649)	2022-02-11 21:30:53 +00:00
partitioner_utils.py
proxy_tensor.py	Add strictness check and made tensors into leaves if input tensors were leaves (#77474)	2022-05-21 01:16:39 +00:00
refinement_types.py	Refinement types and unification for symbolic shape inference (#61776)	2021-08-04 17:34:29 -07:00
rewriter.py	[FX] make ASTReriter patch wrapped functions properly (#62987)	2021-08-18 15:03:57 -07:00
schema_type_annotation.py
unify_refinements.py	inference for algebraic expressions (#63822)	2021-08-25 20:47:23 -07:00