pytorch/torch/fx/passes
Drazen Borkovic f54098cd3e Create JSON from new FX IR and lower to LLVM (#77765)
Summary:
Replace TensorView objects with maps for JSONing.
Lower to LLVM.

Reviewed By: jaybean-dev, jfix71

Differential Revision: D36318989

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77765
Approved by: https://github.com/jfix71, https://github.com/jamesr66a
2022-05-20 03:20:57 +00:00
..
tests [fx][1/2] add PassManager and refactor AFG/AGM (#74972) 2022-04-01 09:12:47 +00:00
__init__.py [FX] Ensure BC coverage for all of torch.fx.passes (#65081) 2021-09-17 09:32:43 -07:00
graph_drawer.py [fx/graph_drawer] Add skip_node_names_in_args option, default to True (#73815) 2022-03-08 01:50:03 +00:00
graph_manipulation.py Create JSON from new FX IR and lower to LLVM (#77765) 2022-05-20 03:20:57 +00:00
net_min_base.py Make args work in the uru10x10_to_trt_eval script (#74707) 2022-03-25 03:52:47 +00:00
operator_support.py Improve fx2trt benchmark (#72145) 2022-02-02 22:41:58 +00:00
param_fetch.py [FX] Ensure BC coverage for all of torch.fx.passes (#65081) 2021-09-17 09:32:43 -07:00
pass_manager.py [fx][1/2] add PassManager and refactor AFG/AGM (#74972) 2022-04-01 09:12:47 +00:00
shape_prop.py [fx][ShapeProp] make shapes and args/kwargs concrete for minimizer (#75291) 2022-04-06 07:57:38 +00:00
split_module.py [FX] Make split_module preserve proper placeholder names (#74736) 2022-03-25 23:36:27 +00:00
split_utils.py Move util functions to a more common place (#73519) 2022-03-01 01:33:30 +00:00
splitter_base.py [FX] Fix type of argument min_acc_module_size (#74891) 2022-03-29 17:48:32 +00:00
tools_common.py Fix fuse_parallel_linear (#76202) 2022-04-22 18:59:07 +00:00