From 49e2a4d85290e75c248e372f16e6c8cb3027a2d8 Mon Sep 17 00:00:00 2001
From: zhiboniu <31800336+zhiboniu@users.noreply.github.com>
Date: Fri, 12 Aug 2022 14:30:33 +0800
Subject: [PATCH] refix index resize in multiclassnms3 (#45095)

---
 paddle/phi/infermeta/ternary.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/phi/infermeta/ternary.cc b/paddle/phi/infermeta/ternary.cc
index 013a58e30ed..0d91f796548 100644
--- 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);
-- 
GitLab