input_tensor,
          id='1',
          calibration_method=_CalibrationMethod.CALIBRATION_METHOD_MIN_MAX,
      )
      aggregator_output = self.evaluate(aggregator)
      self.assertAllEqual(aggregator_output.output, [1.0, 2.0, 3.0, 4.0, 5.0])
      self.assertEqual(aggregator_output.min, 1.0)
      self.assertEqual(aggregator_output.max, 5.0)
      self.assertEmpty(aggregator_output.histogram)

    t3 = constant_op.constant([[1.0, 2.0], [3.0, 4.0]])
    sq1 = add([t1])
    sq2 = add([t1, t2])
    sq3 = add([t1, t2, t3])
    self.assertAllEqual(sq1.numpy().reshape(-1), [1, 2, 3, 4])
    self.assertAllEqual(sq2.numpy().reshape(-1), [2, 4, 6, 8])
    self.assertAllEqual(sq3.numpy().reshape(-1), [3, 6, 9, 12])

  def testBiasedDense(self):
    biased_dense = def_function.function(gen_composite_ops.my_biased_dense)

    sq1 = gen_composite_ops.my_add_n([t1])
    sq2 = gen_composite_ops.my_add_n([t1, t2])
    sq3 = gen_composite_ops.my_add_n([t1, t2, t3])
    self.assertAllEqual(sq1.numpy().reshape(-1), [1, 2, 3, 4])
    self.assertAllEqual(sq2.numpy().reshape(-1), [2, 4, 6, 8])
    self.assertAllEqual(sq3.numpy().reshape(-1), [3, 6, 9, 12])

  def testBiasedDense(self):
    t1 = constant_op.constant([[1.0, 2.0], [3.0, 4.0]])

    statistics.min_max_statistics.global_min = 1.0
    statistics.min_max_statistics.global_max = 5.0

    min_value, max_value = calibration_algorithm.get_min_max_value(
        statistics, calib_opts
    )
    self.assertAllEqual((min_value, max_value), (1.0, 5.0))

  def test_average_min_max(self):
    calib_opts = stablehlo_quant_config_pb2.CalibrationOptions(
        calibration_method=_CalibrationMethod.CALIBRATION_METHOD_AVERAGE_MIN_MAX
    )

    feed_dict = repr_dataset.create_feed_dict_from_input_data(
        sample, signature_def
    )

    self.assertLen(feed_dict, 1)
    self.assertIn('input:0', feed_dict)
    self.assertAllEqual(feed_dict['input:0'], input_tensor_value)

  @test_util.deprecated_graph_mode_only
  def test_create_feed_dict_from_input_data_core_tensors(self):
    signature_def = meta_graph_pb2.SignatureDef(