src/transformers/models/maskformer/modeling_maskformer_swin.py/MaskFormerSwinAttention/__init__
class MaskFormerSwinAttention:    def __init__(self, config, dim, num_heads, window_size):
        super().__init__()
        self.self = MaskFormerSwinSelfAttention(config, dim, num_heads, window_size)
        self.output = MaskFormerSwinSelfOutput(config, dim)
        self.pruned_heads = set()