1 command network inference

include/Network.h

@@ -5,16 +5,31 @@
 
 namespace tkDNN {
 
+struct dataDim_t;
+class Layer;
+const int MAX_LAYERS = 256;
+
 class Network {
 
 public:
     Network();
     virtual ~Network();
 
+    /**
+        Do inferece for every added layer
+    */
+    value_type* infer(dataDim_t &dim, value_type* data);
+
+    bool addLayer(Layer *l);
+
     cudnnDataType_t dataType;
     cudnnTensorFormat_t tensorFormat;
     cudnnHandle_t cudnnHandle;
     cublasHandle_t cublasHandle;
+
+private:
+    Layer* layers[MAX_LAYERS];  //contains layers of the net
+    int num_layers; //current number of layers
 };
 
 }