"""Performs a gather operation."""
        out = array_ops.gather_v2(self.w, input_tensor)
        return {'output': out}

    return GatherModel(use_variable)

  def _create_add_model(
      self,
      shape: Sequence[int],
      saved_model_path: str,
  ) -> module.Module:
    class AddModel(module.Module):
      """A simple model with a single add."""

      def __init__(self):

  )
  @test_util.run_in_graph_and_eager_modes
  def test_add_ptq_model(
      self,
      shape_dynamic: bool,
  ):
    input_shape = (None, 3, 4, 3) if shape_dynamic else (2, 3, 4, 3)
    self._create_add_model(
        input_shape,
        self._input_saved_model_path,
    )

    # Generate model input data.
    rng = np.random.default_rng(seed=42)