[caffegen] Add 'Network' class (#100)
author박종현/동작제어Lab(SR)/Senior Engineer/삼성전자 <jh1302.park@samsung.com>
Thu, 19 Apr 2018 06:40:28 +0000 (15:40 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Thu, 19 Apr 2018 06:40:28 +0000 (15:40 +0900)
This commit adds 'Network' class which serves as in-memory Caffe model
representation.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
contrib/caffegen/include/Network.h [new file with mode: 0644]

diff --git a/contrib/caffegen/include/Network.h b/contrib/caffegen/include/Network.h
new file mode 100644 (file)
index 0000000..7c6eaec
--- /dev/null
@@ -0,0 +1,22 @@
+#ifndef __NETWORK_H__
+#define __NETWORK_H__
+
+#include "BlobContext.h"
+#include "LayerContext.h"
+
+class Network
+{
+public:
+  LayerContext &layers(void) { return _layers; }
+  const LayerContext &layers(void) const { return _layers; }
+
+public:
+  BlobContext &blobs(void) { return _blobs; }
+  const BlobContext &blobs(void) const { return _blobs; }
+
+private:
+  BlobContext _blobs;
+  LayerContext _layers;
+};
+
+#endif // __NETWORK_H__