Searched defs:split_graph (Results 1 – 2 of 2) sorted by relevance
102 def split_graph(edge_program: ExportedProgram, num_layers: int, shares: int): function
448 def split_graph(self, graph_module: torch.fx.GraphModule, division: int): member in TestQNN