src/transformers/models/tapas/modeling_tapas.py/TapasAttention/__init__
class TapasAttention:    def __init__(self, config):
        super().__init__()
        self.self = TapasSelfAttention(config)
        self.output = TapasSelfOutput(config)
        self.pruned_heads = set()