Add support for DisentangledAttentionPlugin (#836)

Signed-off-by: Kevin Chen <[email protected]>
onnx · Apr 28, 2022 · 4ebfd96 · 4ebfd96
1 parent 5f27e45
commit 4ebfd96
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ModelImporter.cpp b/ModelImporter.cpp
@@ -512,7 +512,8 @@ bool ModelImporter::supportsOperator(const char* op_name) const
     {
         return false;
     }
-    if (std::string(op_name) == "EfficientNMS_TRT" || std::string(op_name) == "PyramidROIAlign_TRT" || std::string(op_name) == "MultilevelCropAndResize_TRT")
+    if (std::string(op_name) == "EfficientNMS_TRT" || std::string(op_name) == "PyramidROIAlign_TRT" || std::string(op_name) == "MultilevelCropAndResize_TRT"
+        || std::string(op_name) == "DisentangledAttention_TRT")
     {
         return true;
     }