LEAKY plugin

2017-08-03 13:25:33 +02:00
parent 4526e2767a
commit 2ef76209a1
5 changed files with 79 additions and 12 deletions
@@ -97,6 +97,8 @@ typedef enum {
 class Activation : public Layer {

 public:
+    int act_mode;
+
    Activation(Network *net, int act_mode); 
    virtual ~Activation();
    virtual layerType_t getLayerType() { return LAYER_ACTIVATION; };
@@ -104,7 +106,6 @@ public:
    virtual value_type* infer(dataDim_t &dim, value_type* srcData);

 protected:
-    int act_mode;
    cudnnActivationDescriptor_t activDesc;
 };