Home
last modified time | relevance | path

Searched refs:PatternModel (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/fx/
H A Dtest_matcher_utils.py55 class PatternModel(torch.nn.Module): class
64 pattern_graph = torch.fx.symbolic_trace(PatternModel()).graph
/aosp_15_r20/external/pytorch/docs/source/
H A Dtorch.compiler_transformations.rst353 class PatternModel(torch.nn.Module):
362 pattern_graph = torch.export(PatternModel(), inputs).graph
/aosp_15_r20/external/executorch/docs/source/
H A Dcompiler-custom-compiler-passes.md351 class PatternModel(torch.nn.Module):
360 pattern_graph = to_edge(export(PatternModel(), pattern_inputs)).exported_program().graph_module.gra…