Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/fx/
H A Dtest_fx_const_fold.py63 class ConstFoldTestModule(torch.nn.Module): class
74 mod = ConstFoldTestModule()
108 class ConstFoldTestModule(torch.nn.Module): class
120 mod = ConstFoldTestModule()
135 class ConstFoldTestModule(torch.nn.Module): class
139 mod = ConstFoldTestModule()
170 class ConstFoldTestModule(torch.nn.Module): class
178 mod = ConstFoldTestModule()
213 class ConstFoldTestModule(torch.nn.Module): class
225 mod = ConstFoldTestModule()
[all …]