src/transformers/models/mobilebert/modeling_mobilebert.py/MobileBertAttention/__init__
class MobileBertAttention:    def __init__(self, config):
        super().__init__()
        self.self = MobileBertSelfAttention(config)
        self.output = MobileBertSelfOutput(config)
        self.pruned_heads = set()