code: def get_layer_norm_parameter_tuples():
      """Consists of (input_dims, parameter_axes)."""
      return [
          # Rank-2
          ([3], -1),
          ([3], [1]),
          # Rank-3
          ([3, 4], -1),
          ([3, 4], [1]),
          ([3, 4], [2]),
          ([3, 4], [1, 2]),
          # Rank-4
          ([3, 4, 5], -1),
          ([3, 4, 5], [1]),
          ([3, 4, 5], [2]),
          ([3, 4, 5], [3]),
          ([3, 4, 5], [1, 2]),
          ([3, 4, 5], [1, 3]),
          ([3, 4, 5], [2, 3]),
          ([3, 4, 5], [1, 2, 3]),
      ]
  
file_path: privacy/tensorflow_privacy/privacy/fast_gradient_clipping/registry_functions/layer_normalization_test.py
imports: from absl.testing import parameterized
  import numpy
  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 layer_normalization
label_desc: Use image node_id 3 for calling a global function with example usage: get_layer_norm_parameter_tuples() and returns: list