pytorch/torch/fx/experimental
Huamin Li 32dd4a8639 move fx_acc out of pytorch core (#72803)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72803

as title

Reviewed By: jfix71

Differential Revision: D34101788

fbshipit-source-id: a9fd84671929af21405c049603e9895ec68de3d8
(cherry picked from commit e98fd1c32d)
2022-02-15 16:13:43 +00:00
..
unification Refinement types and unification for symbolic shape inference (#61776) 2021-08-04 17:34:29 -07: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 move fx_acc out of pytorch core (#72803) 2022-02-15 16:13:43 +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 retracing wrapped functions (#58061) 2021-05-17 19:50:16 -07:00
normalize.py [NormalizeArgs] Retain node.meta (#60449) 2021-06-23 03:31:53 -07:00
optimization.py Stop writing logs to root logger (#72649) 2022-02-11 21:30:53 +00:00
partitioner_utils.py [fx-acc] Saturate host by replicating partitions onto idle devices (#60064) 2021-06-15 23:04:46 -07: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 [FX] Experimental type annotation pass using Python signatures (#53831) 2021-03-17 20:43:17 -07:00
unify_refinements.py inference for algebraic expressions (#63822) 2021-08-25 20:47:23 -07:00