torch/fx/passes/ (path) in projects: external

Searched +path:torch +path:fx +path:passes (Results 1 – 25 of 43) sorted by relevance

/external/pytorch/torch/fx/passes/
D	operator_support.py
D	net_min_base.py
D	README.md
D	shape_prop.py
D	graph_transform_observer.py
D	reinplace.py
D	tools_common.py
D	__init__.py
D	graph_manipulation.py
D	runtime_assert.py
D	annotate_getitem_nodes.py
D	splitter_base.py
D	pass_manager.py
D	param_fetch.py
D	fake_tensor_prop.py
D	split_utils.py
D	split_module.py
D	graph_drawer.py
/external/pytorch/torch/fx/passes/tests/
D	__init__.py
D	test_pass_manager.py
/external/pytorch/torch/fx/passes/backends/
D	__init__.py
D	cudagraphs.py
/external/pytorch/torch/fx/passes/utils/
D	__init__.py
D	source_matcher_utils.py
/external/pytorch/torch/fx/passes/infra/
D	__init__.py