Searched refs:MaxPool2dModule (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/backends/xnnpack/test/passes/ |
| H A D | test_remove_get_item_pass.py | 19 class MaxPool2dModule(torch.nn.Module): class in TestRemoveGetItemPass
40 Tester(self.MaxPool2dModule(), (torch.randn(4, 3, 24, 24),))
50 Tester(self.MaxPool2dModule(), (torch.randn(4, 3, 24, 24),))
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A D | test_mps.py | 1689 class MaxPool2dModule(torch.nn.Module): class
1708 maxpool2d_module = MaxPool2dModule(3, 1, 0, 1)
1716 class MaxPool2dModule(torch.nn.Module): class
1736 maxpool2d_module = MaxPool2dModule(3, 1, 0, 1)
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A D | test_vulkan_delegate.py | 522 class MaxPool2dModule(torch.nn.Module): class
537 max_pool2d_module = MaxPool2dModule()
|