refix index resize in multiclassnms3 (#45095)

49e2a4d8 · zhiboniu · GitHub · 5bf3dec9 · 49e2a4d8
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/phi/infermeta/ternary.cc paddle/phi/infermeta/ternary.cc +1 -1

未找到文件。
--- a/paddle/phi/infermeta/ternary.cc
+++ b/paddle/phi/infermeta/ternary.cc
@@ -814,7 +814,7 @@ void MultiClassNMSInferMeta(const MetaTensor& bboxes,

  out->set_dims(phi::make_ddim({-1, box_dims[2] + 2}));
  out->set_dtype(bboxes.dtype());
-  index->set_dims(phi::make_ddim({-1, box_dims[2] + 2}));
+  index->set_dims(phi::make_ddim({-1, 1}));
  index->set_dtype(DataType::INT32);
  nms_rois_num->set_dims(phi::make_ddim({-1}));
  nms_rois_num->set_dtype(DataType::INT32);