code: def get_embedding_inputs():
      """Generates test pairs of the form (is_ragged, input_data)."""
      return [
          # 2D inputs.
          (False, [[0, 1]]),
          (False, [[0, 1], [1, 1], [0, 0]]),
          (True, [[0], [1], [], [0, 0], [0, 1], [1, 0], [1, 1]]),
          (
              True,
              [[0], [1], [], [0, 0], [0, 1], [1, 0], [1, 1], [0, 1]],
          ),
          # 3D inputs.
          (False, [[[0, 1]]]),
          (False, [[[0, 1]], [[1, 1]], [[0, 0]]]),
          (
              True,
              [
                  [[0]],
                  [[1]],
                  [],
                  [[0, 0]],
                  [[0, 1]],
                  [[1, 0]],
                  [[1, 1]],
              ],
          ),
          (
              True,
              [
                  [[0]],
                  [[1]],
                  [],
                  [[0, 0]],
                  [[0, 1]],
                  [[1, 0]],
                  [[1, 1]],
                  [[0]],
              ],
          ),
      ]
  
file_path: privacy/tensorflow_privacy/privacy/fast_gradient_clipping/registry_functions/embedding_test.py
imports: from absl.testing import parameterized
  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 embedding
label_desc: Use image node_id 2 for calling a global function with example usage: get_embedding_inputs() and returns: list