code: def get_einsum_layer_registry():
      einsum_registry = layer_registry.LayerRegistry()
      einsum_registry.insert(
          tfm.nlp.layers.EinsumDense,
          einsum_dense.einsum_layer_computation,
      )
      return einsum_registry
  
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 3 for calling a global function with example usage: get_einsum_layer_registry() and returns: einsum_registry