namespace mlir::quant {

std::unique_ptr<OpQuantSpec> GetUniformOpQuantSpec(Operation* op) {
  auto spec = std::make_unique<OpQuantSpec>();
  if (isa<TF::UniformQuantizedConvolutionHybridOp>(op) ||
      isa<TF::UniformQuantizedConvolutionOp>(op)) {
    spec->coeff_op_quant_dim[1] = 3;
  } else if (isa<TF::UniformQuantizedDotHybridOp>(op)) {
    spec->coeff_op_quant_dim[1] = -1;
  }

  for (auto quantizable_operand : spec->coeff_op_quant_dim) {

      TF::UniformDequantizeOp,
      TF::UniformQuantizeOp,
      TF::UniformQuantizedAddOp,
      TF::UniformQuantizedClipByValueOp,
      TF::UniformQuantizedConvolutionHybridOp,
      TF::UniformQuantizedConvolutionOp,
      TF::UniformQuantizedDotHybridOp,
      TF::UniformQuantizedDotOp,
      TF::UniformRequantizeOp
      // go/keep-sorted end
      // clang-format on
      >(op);
}