// Initialize for tests.
  void initializeForTest() {
    if (!test_mode_) return;

    tensorflow::quantization::QuantizationComponentSpec quant_spec;
    quant_spec.set_quantization_component(
        tensorflow::quantization::QuantizationComponentSpec::COMPONENT_WEIGHT);
    quant_spec.set_tensor_type(
        tensorflow::quantization::QuantizationComponentSpec::TENSORTYPE_INT_8);

            QuantizationComponentSpec* new_spec =
                quant_options_.mutable_quantization_method()
                    ->add_quantization_component_specs();
            new_spec->set_quantization_component(component);
            new_spec->set_tensor_type(type);
          };

      add_new_spec(QuantizationComponentSpec::COMPONENT_ACTIVATION,
                   QuantizationComponentSpec::TENSORTYPE_INT_8);