Searched refs:replace_addmm_with_linear (Results 1 – 1 of 1) sorted by relevance
338 def replace_addmm_with_linear(self, graph_module: torch.fx.GraphModule): member in ReplaceAddMMWithLinearPass410 self.replace_addmm_with_linear(graph_module)