Searched refs:get_graph_module (Results 1 – 4 of 4) sorted by relevance
23 def get_graph_module( member in TestGraphPartition104 graph_module = self.get_graph_module(test_module, example_inputs)136 graph_module = self.get_graph_module(test_module, example_inputs)209 graph_module = self.get_graph_module(test_module, example_inputs)
64 def get_graph_module(self) -> torch.fx.GraphModule: member in GraphBuilder107 return builder.get_graph_module()
458 gm = builder.get_graph_module()500 gm = builder.get_graph_module()536 gm = builder.get_graph_module()577 gm = builder.get_graph_module()
37 gm = builder.get_graph_module()