class_docstr: MultiHeadSelfAttention module optimized for Apple Neural Engine
      
class_docstr_tok: ["MultiHeadSelfAttention","module","optimized","for","Apple","Neural","Engine"]
code: def prune_heads(self, heads):
      raise NotImplementedError
  
imports: from ane_transformers.reference.layer_norm import LayerNormANE
  import torch
  import torch.nn
  from transformers.models.distilbert import modeling_distilbert
label_desc: Use image node_id 2 for calling the MultiHeadSelfAttention obj's underlying member method code with example usage: obj.prune_heads(heads) without return types