support multiclass_nms in int8 (#47337)

eec93bda · zhoutianzi666 · GitHub · 94240e2e · eec93bda
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/inference/tensorrt/op_teller.cc paddle/fluid/inference/tensorrt/op_teller.cc +2 -0

未找到文件。
--- a/paddle/fluid/inference/tensorrt/op_teller.cc
+++ b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -42,6 +42,8 @@ struct SimpleOpTypeSetTeller : public Teller {
    teller_set.insert("group_norm");
    teller_set.insert("multiclass_nms3");
    teller_set.insert("multiclass_nms");
+    int8_teller_set.insert("multiclass_nms3");
+    int8_teller_set.insert("multiclass_nms");
 #endif
 #if IS_TRT_VERSION_GE(7000)
    teller_set.insert("tile");