Home
last modified time | relevance | path

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

/aosp_15_r20/external/executorch/backends/apple/mps/test/
H A Dtest_mps_models.py161 class ModuleReturnInput(torch.nn.Module): class
163 super(ModuleReturnInput, self).__init__()
394 ModuleReturnInput(),
395 ModuleReturnInput().get_random_inputs(),
/aosp_15_r20/external/executorch/test/end2end/
H A Dtest_end2end.py213 class ModuleReturnInput(nn.Module): class
215 super(ModuleReturnInput, self).__init__()