Add support for DisentangledAttentionPlugin (#836)

kevinch-nv · web-flow · commit 4ebfd965edb2 · 2022-04-28T15:19:41.000-07:00
Signed-off-by: Kevin Chen &lt;kevinch@nvidia.com&gt;
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;
     }

Original file line number	Diff line number	Diff line change
`@@ -512,7 +512,8 @@ bool ModelImporter::supportsOperator(const char* op_name) const`
`512`	`512`	`{`
`513`	`513`	`return false;`
`514`	`514`	`}`
`515`		`- if (std::string(op_name) == "EfficientNMS_TRT" \|\| std::string(op_name) == "PyramidROIAlign_TRT" \|\| std::string(op_name) == "MultilevelCropAndResize_TRT")`
	`515`	`+ if (std::string(op_name) == "EfficientNMS_TRT" \|\| std::string(op_name) == "PyramidROIAlign_TRT" \|\| std::string(op_name) == "MultilevelCropAndResize_TRT"`
	`516`	`+ \|\| std::string(op_name) == "DisentangledAttention_TRT")`
`516`	`517`	`{`
`517`	`518`	`return true;`
`518`	`519`	`}`