self.assertAllClose(original_outputs, quantized_output, atol=442.7)

  @test_util.run_v2_only
  def test_while_op_model(
      self,
  ):
    input_shape = (1, 5, 5, 32)
    model = self._create_while_model(input_shape)
    saved_model_save.save(model, self._input_saved_model_path)

    tags = {tag_constants.SERVING}
    quantization_options = quant_opts_pb2.QuantizationOptions(