code: def get_nlp_position_embedding_inputs():
      """Generates input_data."""
      # (input_dims, max_length, seq_axis)
      return [
          # Rank-2 Tensors
          ([3, 2], 6, 1),
          ([3, 2], 3, 1),
          # Rank-3 Tensors
          ([4, 3, 2], 8, 1),
          ([4, 3, 2], 4, 1),
          ([4, 3, 2], 6, 2),
          ([4, 3, 2], 3, 2),
      ]
  
file_path: privacy/tensorflow_privacy/privacy/fast_gradient_clipping/registry_functions/nlp_position_embedding_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 dense
  from tensorflow_privacy.privacy.fast_gradient_clipping.registry_functions import nlp_position_embedding
label_desc: Use image node_id 2 for calling a global function with example usage: get_nlp_position_embedding_inputs() and returns: list