Fix assertion in EfficientNMSPlugin

Documentation states:
```cpp
    // Shape of boxes input should be
    // [batch_size, num_boxes, 4] or [batch_size, num_boxes, 1, 4] or [batch_size, num_boxes, num_classes, 4]
```
however, the second possibility `[batch_size, num_boxes, 1, 4]` triggers a false positive in an
assertion. The third dimension must be either equal to 1 or the number
of classes.

Signed-off-by: Stephan Seitz <[email protected]>

plugin/efficientNMSPlugin/efficientNMSPlugin.cpp

@@ -299,9 +299,9 @@ void EfficientNMSPlugin::configurePlugin(
         }
         else
         {
-            ASSERT(in[0].desc.dims.d[2] == mParam.numClasses);
-            ASSERT(in[0].desc.dims.d[3] == 4);
             mParam.shareLocation = (in[0].desc.dims.d[2] == 1);
+            ASSERT(in[0].desc.dims.d[2] == mParam.numClasses || mParam.shareLocation);
+            ASSERT(in[0].desc.dims.d[3] == 4);
             mParam.numBoxElements = in[0].desc.dims.d[1] * in[0].desc.dims.d[2] * in[0].desc.dims.d[3];
         }
         mParam.numAnchors = in[0].desc.dims.d[1];