Shelfnet works on tensorRT (shortcut need to be fixed)

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>

include/tkDNN/pluginsRT/ActivationLeakyRT.h

@@ -4,9 +4,8 @@
 class ActivationLeakyRT : public IPlugin {
 
 public:
-	ActivationLeakyRT() {
-
-
+	ActivationLeakyRT(float s) {
+		slope = s;
 	}
 
 	~ActivationLeakyRT(){
@@ -42,19 +41,21 @@ public:
 	virtual int enqueue(int batchSize, const void*const * inputs, void** outputs, void* workspace, cudaStream_t stream) override {
 
 		activationLEAKYForward((dnnType*)reinterpret_cast<const dnnType*>(inputs[0]), 
-											reinterpret_cast<dnnType*>(outputs[0]), batchSize*size, stream);
+											reinterpret_cast<dnnType*>(outputs[0]), batchSize*size, slope, stream);
 		return 0;
 	}
 
 
 	virtual size_t getSerializationSize() override {
-		return 1*sizeof(int);
+		return 1*sizeof(int) + 1*sizeof(float);
 	}
 
 	virtual void serialize(void* buffer) override {
 		char *buf = reinterpret_cast<char*>(buffer);
+		tk::dnn::writeBUF(buf, slope);
 		tk::dnn::writeBUF(buf, size);
 	}
 
 	int size;
+	float slope;
 };