Skip to content

Commit 352c5a9

Browse files
author
wanghaox
committed
update some code
2 parents 7bb2680 + 3dcf0da commit 352c5a9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/operators/sequence_slice_op.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,14 +124,14 @@ class SequenceSliceGradOpKernel : public framework::OpKernel<T> {
124124

125125
const int64_t* offset_data = offset->data<int64_t>();
126126
const int64_t* length_data = length->data<int64_t>();
127+
framework::Tensor offset_cpu;
128+
framework::Tensor length_cpu;
127129

128130
if (platform::is_gpu_place(ctx.GetPlace())) {
129-
framework::Tensor offset_cpu;
130131
offset_cpu.mutable_data<T>(offset->dims(), platform::CPUPlace());
131132
offset_cpu.CopyFrom(*offset, platform::CPUPlace(), ctx.device_context());
132133
offset_data = offset_cpu.data<int64_t>();
133134

134-
framework::Tensor length_cpu;
135135
length_cpu.mutable_data<T>(length->dims(), platform::CPUPlace());
136136
length_cpu.CopyFrom(*length, platform::CPUPlace(), ctx.device_context());
137137
length_data = length_cpu.data<int64_t>();

0 commit comments

Comments
 (0)