code: def get_attention_model_generators():
      return {
          "seq1_mha": make_one_layer_attention_model,
          "seq2_mha": make_two_layer_attention_model,
      }
  
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 4 for calling a global function with example usage: get_attention_model_generators() and returns: dict