support CopyToCpu int64_t (#5832)

PaddlePaddle · Apr 6, 2021 · 1cc4dc1 · 1cc4dc1
1 parent c9e2d20
commit 1cc4dc1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lite/api/paddle_api.cc b/lite/api/paddle_api.cc
@@ -210,6 +210,7 @@ template void Tensor::CopyFromCpu<int8_t, TargetType::kMLU>(const int8_t *);
 
 template void Tensor::CopyToCpu(float *) const;
 template void Tensor::CopyToCpu(int *) const;
+template void Tensor::CopyToCpu(int64_t *) const;
 template void Tensor::CopyToCpu(int8_t *) const;
 template void Tensor::CopyToCpu(uint8_t *) const;