diff --git a/paddle/operators/sequence_slice_op.h b/paddle/operators/sequence_slice_op.h index 4ca042b2dd..7210f489e2 100755 --- a/paddle/operators/sequence_slice_op.h +++ b/paddle/operators/sequence_slice_op.h @@ -124,14 +124,14 @@ class SequenceSliceGradOpKernel : public framework::OpKernel { const int64_t* offset_data = offset->data(); const int64_t* length_data = length->data(); + framework::Tensor offset_cpu; + framework::Tensor length_cpu; if (platform::is_gpu_place(ctx.GetPlace())) { - framework::Tensor offset_cpu; offset_cpu.mutable_data(offset->dims(), platform::CPUPlace()); offset_cpu.CopyFrom(*offset, platform::CPUPlace(), ctx.device_context()); offset_data = offset_cpu.data(); - framework::Tensor length_cpu; length_cpu.mutable_data(length->dims(), platform::CPUPlace()); length_cpu.CopyFrom(*length, platform::CPUPlace(), ctx.device_context()); length_data = length_cpu.data();