Searched refs:PatternModel (Results 1 – 3 of 3) sorted by relevance
55 class PatternModel(torch.nn.Module): class64 pattern_graph = torch.fx.symbolic_trace(PatternModel()).graph
353 class PatternModel(torch.nn.Module):362 pattern_graph = torch.export(PatternModel(), inputs).graph
351 class PatternModel(torch.nn.Module):360 pattern_graph = to_edge(export(PatternModel(), pattern_inputs)).exported_program().graph_module.gra…