src/transformers/models/bit/modeling_bit.py/BitMaxPool2d/forward
class BitMaxPool2d:    def forward(self, hidden_states):
        hidden_states = self.pad(hidden_states)
        return nn.functional.max_pool2d(
            hidden_states, self.kernel_size, self.stride, self.padding, self.dilation, self.ceil_mode
        )