src/transformers/models/openai/modeling_tf_openai.py/TFOpenAIGPTMainLayer/_prune_heads
class TFOpenAIGPTMainLayer:    def _prune_heads(self, heads_to_prune):
        """
        Prunes heads of the model. heads_to_prune: dict of {layer_num: list of heads to prune in this layer}
        """
        raise NotImplementedError