|
|
|
@ -33,7 +33,7 @@ class SplitByrefOpKernel : public framework::OpKernel<T> {
|
|
|
|
|
// NOTE: no need to call mutable_data here to allocate memory.
|
|
|
|
|
auto* out = outs[i];
|
|
|
|
|
VLOG(3) << "spliting by ref: " << row_offset << " " << out->dims()[0];
|
|
|
|
|
*out = std::move(in->Slice(row_offset, row_offset + out->dims()[0]));
|
|
|
|
|
*out = in->Slice(row_offset, row_offset + out->dims()[0]);
|
|
|
|
|
row_offset += out->dims()[0];
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|