_set_replace_hook (reference) in projects: aosp_15_r20

Searched refs:_set_replace_hook (Results 1 – 9 of 9) 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()):`
/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	`215 tagged_graph_module._set_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/torch/fx/
H A D	graph_module.py	`896 def _set_replace_hook(self, f): member in GraphModule`