code: def get_attention_layer_registries():
      attention_and_dense = layer_registry.LayerRegistry()
      attention_and_dense.insert(
          tf.keras.layers.MultiHeadAttention,
          multi_head_attention.multi_head_attention_layer_computation,
      )
      attention_and_dense.insert(
          tf.keras.layers.Dense,
          dense.dense_layer_computation,
      )
      return {
          "attention_and_dense": attention_and_dense,
      }
  
imports: from absl.testing import parameterized
  import tensorflow
  from tensorflow_privacy.privacy.fast_gradient_clipping import common_test_utils
  from tensorflow_privacy.privacy.fast_gradient_clipping import layer_registry
  from tensorflow_privacy.privacy.fast_gradient_clipping.registry_functions import dense
  from tensorflow_privacy.privacy.fast_gradient_clipping.registry_functions import multi_head_attention
label_desc: Use image node_id 6 for calling a global function with example usage: get_attention_layer_registries() and returns: dict