return true;
    }
    auto weight_type = input_quantized_types.find(1);
    if (weight_type == input_quantized_types.end()) {
      return true;
    }
    return weight_type->second.has_per_tensor();
  }

  static int GetQuantizationDimension(const WeightOnlyPtq& weight_only_ptq,
                                      TF::XlaCallModuleOp op) {
    const QuantizedType& quantized_type =

      if (static_range_ptq_spec.input_quantized_types().contains(1)) {
        const QuantizedType& weight_quantized_type =
            static_range_ptq_spec.input_quantized_types().at(1);
        if (weight_quantized_type.has_per_tensor()) {
          return false;
        }
        const QuantizedDimension& dimension_specs =
            weight_quantized_type.dimension_specs();
        return !dimension_specs.has_dimension() ||