src/transformers/models/dinat/modeling_dinat.py/NeighborhoodAttentionModule/__init__
class NeighborhoodAttentionModule:    def __init__(self, config, dim, num_heads, kernel_size, dilation):
        super().__init__()
        self.self = NeighborhoodAttention(config, dim, num_heads, kernel_size, dilation)
        self.output = NeighborhoodAttentionOutput(config, dim)
        self.pruned_heads = set()