Searched refs:maxpool_params (Results 1 – 1 of 1) sorted by relevance
120 def __init__(self, maxpool_params): argument122 self.max = torch.nn.MaxPool2d(*maxpool_params)129 for maxpool_params in [(4,), (4, 2), (4, 2, 2)]:132 Tester(MaxPool(maxpool_params), inputs)