stats_op.getResult().replaceAllUsesWith(stats_op.getArg());
    stats_op.erase();
  }

  // Returns false if the steps finish without errors.
  return false;
}

LogicalResult VerifySameScales(Operation* op) {
  auto same_scale_op = cast<SameScalesOpInterface>(op);

  SmallVector<QuantizedType, 4> collected_quant_params;
  for (Value input : op->getOperands()) {
    QuantizedType quant_params =