Value index_to_i64 = rewriter.create<arith::IndexCastOp>(
            op.getLoc(), rewriter.getI64Type(), slice_size_index);
        Value i64_to_tensor = rewriter.create<tensor::FromElementsOp>(
            op.getLoc(),
            tensorflow::GetTypeFromTFTensorShape({1}, rewriter.getI64Type()),
            index_to_i64);
        slice_sizes_values.push_back(i64_to_tensor);