提交 465bf946 编写于 作者: Z zhangjun

update

上级 64e06e45
...@@ -44,12 +44,12 @@ static const int max_batch = 32; ...@@ -44,12 +44,12 @@ static const int max_batch = 32;
static const int min_subgraph_size = 3; static const int min_subgraph_size = 3;
static PrecisionType precision_type; static PrecisionType precision_type;
std::shared_ptr<std::vector<Tensor>> PrepareWarmupData() { std::shared_ptr<std::vector<paddle::PaddleTensor>> PrepareWarmupData() {
auto warmup_data = std::make_shared<std::vector<Tensor>>(1); auto warmup_data = std::make_shared<std::vector<paddle::PaddleTensor>>(1);
Tensor images; paddle::PaddleTensor images;
images.name = "image"; images.name = "image";
images.shape = {2, 3, 300, 300}; images.shape = {2, 3, 300, 300};
images.dtype = PrecisionType::kFloat32; images.dtype = paddle::PaddleDType::FLOAT32;
images.data.Resize(sizeof(float) * 2 * 3 * 300 * 300); images.data.Resize(sizeof(float) * 2 * 3 * 300 * 300);
(*warmup_data)[0] = std::move(images); (*warmup_data)[0] = std::move(images);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册