Searched refs:SliceModule (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nnapi.py | 196 class SliceModule(torch.nn.Module): class
210 self.check(SliceModule(1, 5, 2), torch.randn(4, 6, 2))
215 SliceModule(1, 5, 2),
221 SliceModule(1, 5, 2),
|