p << ' ';
  auto operands = getOperands();
  // Print the 2 data operands.
  p.printOperands(operands.begin(), std::next(operands.begin(), 2));
  p << " of " << (getNumResults() - 1);
  // print control dependencies if any
  if (!getControlInputs().empty()) {
    p << " (";
    p.printOperands(getControlInputs());
    p << ")";
  }
  p << " : " << getType(0);

      if (block_arg_num < num_replicated_block_args) {
        p << '[';
        p.printOperands(
            std::next(getReplicatedInputs().begin(), block_arg_num * n),
            std::next(getReplicatedInputs().begin(), (block_arg_num + 1) * n));
        p << "]";
      } else {
        p.printOperand(*std::next(getPackedInputs().begin(),
                                  block_arg_num - num_replicated_block_args));

  if (llvm::any_of(op->getOperandTypes(),
                   [&](Type type) { return type != resultType; })) {
    p.printGenericOp(op, /*printOpName=*/false);
    return;
  }

  p << ' ';
  p.printOperands(op->getOperands());
  p.printOptionalAttrDict(op->getAttrs());
  // Now we can output only one type for all operands and the result.
  p << " : " << resultType;
}