code: def get_attention_layer_generators():
      def basic_attention_layer(
          num_heads,
          key_dim,
          value_dim,
          dropout,
          use_bias,
          output_shape,
      ):
          return tf.keras.layers.MultiHeadAttention(
              num_heads,
              key_dim,
              value_dim=value_dim,
              dropout=dropout,
              use_bias=use_bias,
              output_shape=output_shape,
          )
  
      return {
          "basic_attention_layer": basic_attention_layer,
      }
  
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 1 for calling a global function with example usage: get_attention_layer_generators() and returns: dict, tf