get_replace_hook (reference) in projects: aosp_15_r20

Searched refs:get_replace_hook (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/export/
H A D	_remove_auto_functionalized_pass.py	`34 with ep.graph_module._set_replace_hook(ep.graph_signature.get_replace_hook()):`
H A D	_remove_effect_tokens_pass.py	`156 with ep.graph_module._set_replace_hook(ep.graph_signature.get_replace_hook()):`
H A D	exported_program.py	`562 with gm._set_replace_hook(new_graph_signature.get_replace_hook()): 584 with gm._set_replace_hook(graph_signature.get_replace_hook()):`
H A D	graph_signature.py	`438 def get_replace_hook(self): member in ExportGraphSignature`
/aosp_15_r20/external/pytorch/torch/export/experimental/
H A D	__init__.py	`35 with gm._set_replace_hook(sig.get_replace_hook()):`
/aosp_15_r20/external/pytorch/torch/_export/passes/
H A D	replace_with_hop_pass_util.py	`136 …replace_ctx = new_gm._set_replace_hook(new_signature.get_replace_hook()) # type: ignore[assignmen…`
/aosp_15_r20/external/pytorch/test/export/
H A D	test_pass_infra.py	`184 with gm._set_replace_hook(sig.get_replace_hook()):`
/aosp_15_r20/external/executorch/exir/backend/
H A D	backend_api.py	`216 owning_program.graph_signature.get_replace_hook()`
/aosp_15_r20/external/pytorch/torch/distributed/tensor/experimental/
H A D	_tp_transform.py	`53 with gm._set_replace_hook(sig.get_replace_hook()):`
/aosp_15_r20/external/pytorch/docs/source/
H A D	export.rst	`711 .. automethod:: get_replace_hook`