diff --git a/src/pluginsRT/ActivationMishRT.cpp b/src/pluginsRT/ActivationMishRT.cpp index d42b90a..f92629b 100644 --- a/src/pluginsRT/ActivationMishRT.cpp +++ b/src/pluginsRT/ActivationMishRT.cpp @@ -73,7 +73,7 @@ const char *ActivationMishRT::getPluginVersion() const NOEXCEPT { } bool ActivationMishRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } const char *ActivationMishRT::getPluginNamespace() const NOEXCEPT { diff --git a/src/pluginsRT/DeformableConvRT.cpp b/src/pluginsRT/DeformableConvRT.cpp index 9b7222d..6af3587 100644 --- a/src/pluginsRT/DeformableConvRT.cpp +++ b/src/pluginsRT/DeformableConvRT.cpp @@ -287,7 +287,7 @@ void DeformableConvRT::detachFromContext() NOEXCEPT { } bool DeformableConvRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } DeformableConvRTPluginCreator::DeformableConvRTPluginCreator() { diff --git a/src/pluginsRT/FlattenConcatRT.cpp b/src/pluginsRT/FlattenConcatRT.cpp index 53036ff..0e500b3 100644 --- a/src/pluginsRT/FlattenConcatRT.cpp +++ b/src/pluginsRT/FlattenConcatRT.cpp @@ -158,7 +158,7 @@ FlattenConcatRT::configurePlugin(const Dims *inputDims, int32_t nbInputs, const } bool FlattenConcatRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } FlattenConcatRTPluginCreator::FlattenConcatRTPluginCreator() { diff --git a/src/pluginsRT/MaxPoolingSizeRT.cpp b/src/pluginsRT/MaxPoolingSizeRT.cpp index 5f36ae3..c45eae3 100644 --- a/src/pluginsRT/MaxPoolingSizeRT.cpp +++ b/src/pluginsRT/MaxPoolingSizeRT.cpp @@ -93,8 +93,7 @@ delete this; } bool MaxPoolFixedSizeRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; - //todo assert + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } const char *MaxPoolFixedSizeRT::getPluginNamespace() const NOEXCEPT { diff --git a/src/pluginsRT/RegionRT.cpp b/src/pluginsRT/RegionRT.cpp index 249c48a..c31bbe6 100644 --- a/src/pluginsRT/RegionRT.cpp +++ b/src/pluginsRT/RegionRT.cpp @@ -131,7 +131,7 @@ void RegionRT::setPluginNamespace(const char *pluginNamespace) NOEXCEPT { } bool RegionRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } IPluginV2Ext *RegionRT::clone() const NOEXCEPT { diff --git a/src/pluginsRT/ReorgRT.cpp b/src/pluginsRT/ReorgRT.cpp index e3a18df..8f0fa12 100644 --- a/src/pluginsRT/ReorgRT.cpp +++ b/src/pluginsRT/ReorgRT.cpp @@ -75,7 +75,7 @@ void ReorgRT::serialize(void *buffer) const NOEXCEPT { } bool ReorgRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } const char *ReorgRT::getPluginType() const NOEXCEPT { diff --git a/src/pluginsRT/ReshapeRT.cpp b/src/pluginsRT/ReshapeRT.cpp index f026d3a..8d81a8d 100644 --- a/src/pluginsRT/ReshapeRT.cpp +++ b/src/pluginsRT/ReshapeRT.cpp @@ -76,8 +76,7 @@ void ReshapeRT::serialize(void *buffer) const NOEXCEPT { } bool ReshapeRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; - //todo assert + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } const char *ReshapeRT::getPluginType() const NOEXCEPT { diff --git a/src/pluginsRT/ResizeLayerRT.cpp b/src/pluginsRT/ResizeLayerRT.cpp index db98326..1068f60 100644 --- a/src/pluginsRT/ResizeLayerRT.cpp +++ b/src/pluginsRT/ResizeLayerRT.cpp @@ -77,8 +77,7 @@ void ResizeLayerRT::serialize(void *buffer) const NOEXCEPT { } bool ResizeLayerRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; - //todo assert + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } const char *ResizeLayerRT::getPluginType() const NOEXCEPT { diff --git a/src/pluginsRT/RouteRT.cpp b/src/pluginsRT/RouteRT.cpp index 3218622..3c3ca79 100644 --- a/src/pluginsRT/RouteRT.cpp +++ b/src/pluginsRT/RouteRT.cpp @@ -131,8 +131,7 @@ void RouteRT::setPluginNamespace(const char *pluginNamespace) NOEXCEPT { } bool RouteRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; - //todo assert + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } IPluginV2 *RouteRT::clone() const NOEXCEPT { diff --git a/src/pluginsRT/ShortcutRT.cpp b/src/pluginsRT/ShortcutRT.cpp index 2a126bb..606f66e 100644 --- a/src/pluginsRT/ShortcutRT.cpp +++ b/src/pluginsRT/ShortcutRT.cpp @@ -91,7 +91,7 @@ void ShortcutRT::serialize(void *buffer) const NOEXCEPT { } bool ShortcutRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } const char *ShortcutRT::getPluginType() const NOEXCEPT { diff --git a/src/pluginsRT/UpsampleRT.cpp b/src/pluginsRT/UpsampleRT.cpp index 21ee95a..af21bc0 100644 --- a/src/pluginsRT/UpsampleRT.cpp +++ b/src/pluginsRT/UpsampleRT.cpp @@ -81,7 +81,7 @@ void UpsampleRT::serialize(void *buffer) const NOEXCEPT { } bool UpsampleRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } const char *UpsampleRT::getPluginType() const NOEXCEPT { diff --git a/src/pluginsRT/YoloRT.cpp b/src/pluginsRT/YoloRT.cpp index 4ef62b7..9585754 100644 --- a/src/pluginsRT/YoloRT.cpp +++ b/src/pluginsRT/YoloRT.cpp @@ -152,7 +152,7 @@ size_t YoloRT::getSerializationSize() const NOEXCEPT { } bool YoloRT::supportsFormat(DataType type, PluginFormat format) const NOEXCEPT { - return true; + return (type == DataType::kFLOAT && format == PluginFormat::kLINEAR); } void YoloRT::serialize(void *buffer) const NOEXCEPT {