class_docstr: Unit tests for the ClusteringRegistry class.
class_docstr_tok: ["Unit","tests","for","the","ClusteringRegistry","class","."]
docstr: Verifies that ClusterRegistry supports a MultiHeadAttention layer.
docstr_tok: ["Verifies","that","ClusterRegistry","supports","a","MultiHeadAttention","layer","."]
code: def testSupportsMultiHeadAttentionLayer(self):
      """Verifies that ClusterRegistry supports a MultiHeadAttention layer."""
      layer = tf.keras.layers.MultiHeadAttention(num_heads=2, key_dim=2)
      self.assertTrue(ClusterRegistry.supports(layer))
      ClusterRegistry.make_clusterable(layer)
  
imports: from absl.testing import parameterized
  import tensorflow
  from tensorflow_model_optimization.python.core.clustering.keras import clusterable_layer
  from tensorflow_model_optimization.python.core.clustering.keras import clustering_registry
  from tensorflow_model_optimization.python.core.clustering.keras.cluster_config import GradientAggregation
label_desc: Use image node_id 21 for calling the ClusterRegistryTest obj's underlying member method code with example usage: obj.testSupportsMultiHeadAttentionLayer() without return types