add matrix_nms extension

liubo-intel · Jul 1, 2021 · 879bb18 · 879bb18
1 parent 6182445
commit 879bb18
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/inference-engine/src/mkldnn_plugin/nodes/list_tbl.hpp b/inference-engine/src/mkldnn_plugin/nodes/list_tbl.hpp
@@ -6,3 +6,5 @@
 # warning "MKLDNN_EXTENSION_NODE is not defined"
 # define MKLDNN_EXTENSION_NODE(__prim, __type)
 #endif
+
+MKLDNN_EXTENSION_NODE(MatrixNmsImpl, MatrixNmsIEInternal);