test=develop test=xpu

PaddlePaddle · Jul 21, 2020 · 2e00379 · 2e00379
1 parent e0c42aa
commit 2e00379
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lite/backends/arm/math/beam_search.cc b/lite/backends/arm/math/beam_search.cc
@@ -234,7 +234,7 @@ void beam_search(const Tensor *pre_ids,
   selected_ids->Resize(dims);
   selected_scores->Resize(dims);
   if (parent_idx) {
-    parent_idx->Resize(std::vector<int64_t>({static_cast<int>(num_instances)}));
+    parent_idx->Resize({static_cast<int64_t>(num_instances)});
   }
   auto *selected_ids_data = selected_ids->mutable_data<int64_t>();
   auto *selected_scores_data = selected_scores->mutable_data<float>();