LogicalResult matchAndRewrite(TF::SliceOp op,
                                PatternRewriter &rewriter) const override {
    Location loc = op.getLoc();
    Value input = op.getInput();
    Value begin_indices = op.getBegin();
    Value sizes = op.getSize();

    auto input_ty = mlir::dyn_cast<RankedTensorType>(input.getType());
    auto begin_type = mlir::dyn_cast<RankedTensorType>(begin_indices.getType());