Searched refs:fx_graph_cse (Results 1 – 5 of 5) sorted by relevance
4 from torch._functorch.compile_utils import fx_graph_cse27 new_g = fx_graph_cse(fx_g.graph)
13 from torch._functorch.compile_utils import fx_graph_cse204 new_graph = fx_graph_cse(fx_g.graph)220 pass_2_graph = fx_graph_cse(new_graph)
13 from torch._functorch.compile_utils import fx_graph_cse104 cse_graph = fx_graph_cse(aot_autograd_gm.graph)
42 def fx_graph_cse(fx_g: torch.fx.graph.Graph): function
33 from .compile_utils import fx_graph_cse, get_aten_target1768 cse_graph = fx_graph_cse(fx_g)