return failure();
    }
    rewriter.replaceOpWithNewOp<TF::CastOp>(op, op.getResult(0).getType(),
                                            input);
    return success();
  }
};

class HandleOptionalNone : public OpRewritePattern<TF::OptionalNoneOp> {
 public:
  using OpRewritePattern::OpRewritePattern;
  LogicalResult matchAndRewrite(TF::OptionalNoneOp op,