Enable DeBERTa plugin in parser

symphonylyh · Apr 20, 2022 · 0510e54 · 0510e54
1 parent 5f27e45
commit 0510e54
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ModelImporter.cpp b/ModelImporter.cpp
@@ -516,6 +516,10 @@ bool ModelImporter::supportsOperator(const char* op_name) const
     {
         return true;
     }
+    if (std::string(op_name) == "DisentangledAttentionPlugin")
+    {
+        return true;
+    }
     return _op_importers.count(op_name);
 }
 bool ModelImporter::parseWithWeightDescriptors(void const* serialized_onnx_model, size_t serialized_onnx_model_size)