include/tkDNN/pluginsRT/ActivationLogisticRT.h

@@ -70,7 +70,7 @@ public:
 
     // Extra IPluginV2 overrides
     bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-        return true;
+        return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
     }
 
     nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/ActivationMishRT.h

@@ -71,7 +71,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/ActivationReLUCeilingRT.h

@@ -75,7 +75,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/ActivationSigmoidRT.h

@@ -70,7 +70,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/DeformableConvRT.h

@@ -185,7 +185,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/FlattenConcatRT.h

@@ -93,7 +93,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/MaxPoolingFixedSizeRT.h

@@ -86,7 +86,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/RegionRT.h

@@ -99,7 +99,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/ReorgRT.h

@@ -76,7 +76,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/ReshapeRT.h

@@ -78,7 +78,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/ResizeLayerRT.h

@@ -83,7 +83,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/RouteRT.h

@@ -105,7 +105,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/ShortcutRT.h

@@ -89,7 +89,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/UpsampleRT.h

@@ -78,7 +78,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {

include/tkDNN/pluginsRT/YoloRT.h

@@ -137,7 +137,7 @@ public:
 
 	// Extra IPluginV2 overrides
 	bool supportsFormat(nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override {
-			return true;
+			return (type == nvinfer1::DataType::kFLOAT && format == nvinfer1::PluginFormat::kLINEAR);
 	}
 
 	nvinfer1::IPluginV2 * clone() const noexcept override {