ASSERT_THAT(srq_spec.input_quantized_types(), SizeIs(1));
  ASSERT_TRUE(srq_spec.input_quantized_types().contains(1));
  ASSERT_TRUE(srq_spec.input_quantized_types().at(1).has_dimension_specs());

  const QuantizedDimension& dimension_specs =
      srq_spec.input_quantized_types().at(1).dimension_specs();
  ASSERT_TRUE(dimension_specs.has_dimension());
  EXPECT_THAT(dimension_specs.dimension(), Eq(3));

    // populate `coeff_op_quant_dim`.
    for (const auto& [operand_idx, quantized_type] :
         static_range_ptq_spec.input_quantized_types()) {
      if (quantized_type.has_dimension_specs()) {
        spec.coeff_op_quant_dim[operand_idx] =
            quantized_type.dimension_specs().dimension();
      }
    }
  }
}

}  // namespace