average_min_max_mse = get_mean_square_error(
        original_output, average_min_max_output
    )

    self.assertLess(average_min_max_mse, min_max_mse)


class SelectiveQuantizationTest(quantize_model_test_base.QuantizedModelTest):
  """Test cases regarding selective quantization."""

  def test_unitwise_spec_with_no_units(self):
    _, y_shape, bias_shape, x_signature, y_signature = (