diff --git a/paddle/fluid/inference/lite/tensor_utils.cc b/paddle/fluid/inference/lite/tensor_utils.cc index 8e7ebd3119e976555b6281a73601723f48dcf12a..6138e64e2db376a45ede4f13b3fb9df7d6e03461 100644 --- a/paddle/fluid/inference/lite/tensor_utils.cc +++ b/paddle/fluid/inference/lite/tensor_utils.cc @@ -166,10 +166,10 @@ void TensorCopyAsync(paddle::lite::Tensor* dst, const framework::LoDTensor& src, template <> void TensorCopyAsync(framework::LoDTensor* dst, const paddle::lite::Tensor& src, const platform::DeviceContext& ctx) { + dst->Resize(paddle::framework::make_ddim(src.dims().Vectorize())); InitDstTensor(dst, src); const platform::Place& src_place = GetNativePlace(src.target()); const platform::Place& dst_place = dst->place(); - dst->Resize(paddle::framework::make_ddim(src.dims().Vectorize())); const size_t bytes = static_cast(src.numel()) * framework::SizeOfType(dst->type()); const void* src_data = src.raw_data();