src/transformers/models/swin/modeling_tf_swin.py/TFSwinAttention/prune_heads
class TFSwinAttention:    def prune_heads(self, heads):
        """
        Prunes heads of the model. See base class PreTrainedModel heads: dict of {layer_num: list of heads to prune in
        this layer}
        """
        raise NotImplementedError