Lines Matching defs:GraphModule

288         def __init__(self, callback: "_ExportPassBase", gm: fx.GraphModule) -> None:
417     def inputs(self, graph_module: torch.fx.GraphModule) -> List[Argument]:
483         true_fn: torch.fx.GraphModule,
484         false_fn: torch.fx.GraphModule,
502         f: torch.fx.GraphModule,
527         self, graph_module: fx.GraphModule, inputs: Tuple[Argument, ...]
550     def call(self, graph_module: fx.GraphModule) -> PassResult:
597         def __init__(self, callback: "ExportPass", gm: fx.GraphModule) -> None:
660         self, graph_module: fx.GraphModule, inputs: Tuple[Argument, ...]
665             gm: torch.fx.GraphModule, new_gm: torch.fx.GraphModule
667             def get_phs(gm: torch.fx.GraphModule) -> List[torch.fx.Node]: