Home
last modified time | relevance | path

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

/aosp_15_r20/external/executorch/backends/apple/mps/test/
H A Dtest_mps_models.py74 class ModuleFloatAddWithAlpha(torch.nn.Module): class
76 super(ModuleFloatAddWithAlpha, self).__init__()
370 ModuleFloatAddWithAlpha(),
371 ModuleFloatAddWithAlpha().get_random_inputs(),
/aosp_15_r20/external/executorch/test/end2end/
H A Dtest_end2end.py126 class ModuleFloatAddWithAlpha(nn.Module): class
128 super(ModuleFloatAddWithAlpha, self).__init__()