code: def get_einsum_layer_generators():
      def pure_einsum_layer(equation, output_dims, bias_axes):
          return tf.keras.layers.EinsumDense(
              equation, output_dims, bias_axes=bias_axes
          )
  
      def sigmoid_einsum_layer(equation, output_dims, bias_axes):
          return tf.keras.layers.EinsumDense(
              equation,
              output_dims,
              bias_axes=bias_axes,
              activation="sigmoid",
          )
  
      return {
          "pure_einsum": pure_einsum_layer,
          "sigmoid_einsum": sigmoid_einsum_layer,
      }
  
file_path: privacy/tensorflow_privacy/privacy/fast_gradient_clipping/registry_functions/einsum_dense_test.py
imports: from absl.testing import parameterized
  import tensorflow
  import tensorflow_models
  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 einsum_dense
label_desc: Use image node_id 1 for calling a global function with example usage: get_einsum_layer_generators() and returns: dict, tf, tf