Home
last modified time | relevance | path

Searched refs:ModuleForwardHasGraphBreak (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/dynamo/pr_time_benchmarks/benchmarks/
H A Dbasic_modules_benchmarks.py33 class ModuleForwardHasGraphBreak(torch.nn.Module): class
150 Benchmark(ModuleForwardHasGraphBreak, "inductor"),
151 Benchmark(ModuleForwardHasGraphBreak, "eager"),
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_modules.py978 class ModuleForwardHasGraphBreak(torch.nn.Module): class
1220 m = ModuleForwardHasGraphBreak()