Added initial support for Layers

bnn/nn/activations.hpp

@@ -0,0 +1,56 @@
+#ifndef BNN_BNN_NN_ACTIVATIONS_HPP
+#define BNN_BNN_NN_ACTIVATIONS_HPP
+
+#include <bnn/utils/utils.hpp>
+#include <bnn/operations/operators.hpp>
+
+namespace bnn
+{
+    namespace nn
+    {
+
+        using namespace bnn::utils;
+        using namespace bnn::operators;
+
+        template <class data_type>
+        class Activation: public BNNBase
+        {
+
+            protected:
+
+                std::string name;
+
+                Operator<data_type>* input;
+
+                Operator<data_type>* output;
+
+            public:
+
+                Activation
+                (std::string _name);
+
+                virtual
+                void
+                implementation
+                ();
+
+                virtual
+                TensorCPU<data_type>*
+                get_input_value
+                ();
+
+                virtual
+                TensorCPU<data_type>*
+                get_output_value
+                ();
+
+                virtual
+                ~Activation
+                ();
+
+        };
+
+    }
+}
+
+#endif

bnn/nn/layers.hpp

@@ -0,0 +1,69 @@
+#ifndef BNN_BNN_NN_LAYERS_HPP
+#define BNN_BNN_NN_LAYERS_HPP
+
+#include <bnn/utils/utils.hpp>
+#include <bnn/core/tensor.hpp>
+#include <bnn/operations/operators.hpp>
+#include <string>
+
+namespace bnn
+{
+    namespace nn
+    {
+
+        using namespace bnn::utils;
+        using namespace bnn::core;
+        using namespace bnn::operators;
+
+        template <class data_type>
+        class Layer: public BNNBase
+        {
+
+            protected:
+
+                std::string name;
+
+                Operator<data_type>* input;
+
+                Operator<data_type>* output;
+
+            public:
+
+                Layer
+                (std::string _name);
+
+                virtual
+                void
+                initialize_parameters
+                (void (*initializer)(Operator<data_type>* parameters));
+
+                virtual
+                void
+                implementation
+                ();
+
+                virtual
+                Operator<data_type>*
+                get_parameters
+                ();
+
+                virtual
+                TensorCPU<data_type>*
+                get_input_value
+                ();
+
+                virtual
+                TensorCPU<data_type>*
+                get_output_value
+                ();
+
+                virtual
+                ~Layer
+                ();
+
+        };
+
+    }
+}
+
+#endif

bnn/operations/operators.hpp

@@ -2,7 +2,6 @@
 #define BNN_BNN_OPERATIONS_OPERATORS_HPP
 
 #include <string>
-#include <unordered_map>
 #include <bnn/core/tensor.hpp>
 #include <bnn/utils/utils.hpp>