code: def get_nlp_on_device_embedding_inputs():
      """Generates input_data."""
      return [
          # 2D inputs.
          [[0, 1]],
          [[0, 1], [1, 1], [0, 0]],
          # 3D inputs.
          [[[0, 1]]],
          [[[0, 1]], [[1, 1]], [[0, 0]]],
      ]
  
file_path: privacy/tensorflow_privacy/privacy/fast_gradient_clipping/registry_functions/nlp_on_device_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_on_device_embedding
label_desc: Use image node_id 2 for calling a global function with example usage: get_nlp_on_device_embedding_inputs() and returns: list