src/transformers/models/nezha/modeling_nezha.py/NezhaAttention/__init__
class NezhaAttention:    def __init__(self, config):
        super().__init__()
        self.self = NezhaSelfAttention(config)
        self.output = NezhaSelfOutput(config)
        self.pruned_heads = set()