Searched refs:replace_add_with_mul (Results 1 – 2 of 2) sorted by relevance
92 def replace_add_with_mul(gm: torch.fx.GraphModule) -> None: function112 pm = PassManager(passes=[replace_add_with_mul, replace_mul_with_div])
23 def replace_add_with_mul(gm: torch.fx.GraphModule) -> torch.fx.GraphModule: