diff --git a/paddle/operators/math/im2col.cc b/paddle/operators/math/im2col.cc index d3a736a62d3ee430e7f5005fd655770b9d3a5f3d..3b1b0bd71dd3768b932864e185af8dc839b4653e 100644 --- a/paddle/operators/math/im2col.cc +++ b/paddle/operators/math/im2col.cc @@ -282,7 +282,7 @@ class Col2ImFunctor(input_tmp, *place, *context); + input.CopyFrom(input_tmp, *place, *context); } col2im(*context, input, output_cfo, stride, stride, padding, padding, padding, @@ -138,7 +138,7 @@ void testIm2col() { if (paddle::platform::is_cpu_place(*place)) { in_ptr = input.data(); } else { - input_tmp.CopyFrom(input, paddle::platform::CPUPlace(), *context); + input_tmp.CopyFrom(input, paddle::platform::CPUPlace(), *context); in_ptr = input_tmp.data(); } for (int i = 0; i < 6; ++i) { @@ -150,7 +150,7 @@ void testIm2col() { if (paddle::platform::is_cpu_place(*place)) { input = input_tmp; } else { - input.CopyFrom(input_tmp, *place, *context); + input.CopyFrom(input_tmp, *place, *context); } col2im_ocf(*context, input, output_ocf, stride, stride, padding, padding, @@ -159,7 +159,7 @@ void testIm2col() { if (paddle::platform::is_cpu_place(*place)) { in_ptr = input.data(); } else { - input_tmp.CopyFrom(input, paddle::platform::CPUPlace(), *context); + input_tmp.CopyFrom(input, paddle::platform::CPUPlace(), *context); in_ptr = input_tmp.data(); } for (int i = 0; i < 6; ++i) {