Searched refs:NNPoolingModule (Results 1 – 2 of 2) sorted by relevance
24 NNPoolingModule,75 "pooling_ops": NNPoolingModule(),
44 class NNPoolingModule(torch.nn.Module): class