Example #1
0
 def __init__(self,
              kernel_size,
              stride=None,
              padding=0,
              dilation=1,
              return_indices=False,
              ceil_mode=False,
              return_quant_tensor: bool = True):
     MaxPool2d.__init__(self,
                        kernel_size=kernel_size,
                        stride=stride,
                        padding=padding,
                        dilation=dilation,
                        return_indices=return_indices,
                        ceil_mode=ceil_mode)
     QuantLayerMixin.__init__(self, return_quant_tensor=return_quant_tensor)