auto then_type =
        mlir::dyn_cast<RankedTensorType>(op.getThenValue().getType());
    auto else_type =
        mlir::dyn_cast<RankedTensorType>(op.getElseValue().getType());
    if (!cond_type || !then_type || !else_type) {
      return failure();
    }

    ImplicitLocOpBuilder b(op.getLoc(), rewriter);

//         first dimension equal to `cond`.
LogicalResult SelectOp::verify() {
  SelectOp op = *this;
  auto then_tensor = op.getThenValue().getType().cast<TensorType>();
  auto else_tensor = op.getElseValue().getType().cast<TensorType>();
  // Check (1).
  if (!AreCastCompatible({then_tensor, else_tensor}))
    return op.emitOpError() << "requires t and e have compatible shapes";

  // Get data rank (if exists).