initial commit

2017-06-28 01:14:39 +02:00
commit 0767df43a2
12 changed files with 441 additions and 0 deletions
@@ -0,0 +1,21 @@
+#ifndef NETWORK_H
+#define NETWORK_H
+
+#include "utils.h"
+
+namespace tkDNN {
+
+class Network {
+
+public:
+    Network();
+    virtual ~Network();
+
+    cudnnDataType_t dataType;
+    cudnnTensorFormat_t tensorFormat;
+    cudnnHandle_t cudnnHandle;
+    cublasHandle_t cublasHandle;
+};
+
+}
+#endif //NETWORK_H