Merge pull request #11 from luo-cheng2021/yi3/add_mkl_matrixnms

add matrix_nms extension
liubo-intel · Jul 1, 2021 · 769a3c9 · 769a3c9
2 parents 6182445 + 879bb18
commit 769a3c9
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);