diff --git a/paddle_inference/paddle/include/paddle_engine.h b/paddle_inference/paddle/include/paddle_engine.h index 1699487473ca490e1c0a1a2cf57aeb86c261bb4c..58144e7da769cb144def73c396aa1f0c04543fb2 100755 --- a/paddle_inference/paddle/include/paddle_engine.h +++ b/paddle_inference/paddle/include/paddle_engine.h @@ -44,12 +44,12 @@ static const int max_batch = 32; static const int min_subgraph_size = 3; static PrecisionType precision_type; -std::shared_ptr> PrepareWarmupData() { - auto warmup_data = std::make_shared>(1); - Tensor images; +std::shared_ptr> PrepareWarmupData() { + auto warmup_data = std::make_shared>(1); + paddle::PaddleTensor images; images.name = "image"; images.shape = {2, 3, 300, 300}; - images.dtype = PrecisionType::kFloat32; + images.dtype = paddle::PaddleDType::FLOAT32; images.data.Resize(sizeof(float) * 2 * 3 * 300 * 300); (*warmup_data)[0] = std::move(images);