From 4ebfd965edb2737d02c3202749a7bc499d53c585 Mon Sep 17 00:00:00 2001 From: Kevin Chen <45886021+kevinch-nv@users.noreply.github.com> Date: Thu, 28 Apr 2022 15:19:41 -0700 Subject: [PATCH] Add support for DisentangledAttentionPlugin (#836) Signed-off-by: Kevin Chen --- ModelImporter.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ModelImporter.cpp b/ModelImporter.cpp index f5a147a6..01bff224 100644 --- 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; }