[ NNSTREAMER FILTER ] Fix nnstreamer filter to support graph
authorjijoong.moon <jijoong.moon@samsung.com>
Fri, 20 Nov 2020 00:58:43 +0000 (09:58 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Thu, 26 Nov 2020 07:27:55 +0000 (16:27 +0900)
Describe a commit content (Until 80 colums per line) in detail ASAP.

**Changes proposed in this PR:**
- Added TOC generator for README.md

Resolves:

**Self evaluation:**
1. Build test:  [X]Passed [ ]Failed [ ]Skipped
2. Run test:  [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: jijoong.moon <jijoong.moon@samsung.com>
nnstreamer/tensor_filter/tensor_filter_nntrainer.cc
nntrainer/layers/layer_internal.h

index 9eaf94ffec89840b38294040261b28df807acf4b..8db29e7f0c2dd1bcbc1091901df454a57efe9629 100644 (file)
@@ -91,7 +91,7 @@ NNTrainer::NNTrainer(const char *model_config_,
   validateTensor(&prop->input_meta, true);
   validateTensor(&prop->output_meta, false);
 
-  model->init();
+  model->initialize();
   model->readModel();
 
   gst_tensors_info_copy(&inputTensorMeta, &prop->input_meta);
@@ -147,6 +147,7 @@ void NNTrainer::loadModel() {
 
   model = new nntrainer::NeuralNetwork();
   model->loadFromConfig(model_config);
+  model->compile();
 #if (DBG)
   gint64 stop_time = g_get_real_time();
   g_message("Model is loaded: %" G_GINT64_FORMAT, (stop_time - start_time));
index 9579e028a97b2cc8fe66e9897e9e60b0a07e8f0d..47517f821c21e1db36e5edc4ddbd004e2d5e2253 100644 (file)
@@ -114,7 +114,7 @@ public:
   virtual void forwarding(sharedConstTensors in = {}) = 0;
 
   virtual sharedConstTensors forwarding_with_val(sharedConstTensors input,
-                                      sharedConstTensors in = {});
+                                                 sharedConstTensors in = {});
 
   /**
    * @brief     Back Propagation of a layer
@@ -124,9 +124,9 @@ public:
    */
   virtual void backwarding(int iteration, sharedConstTensors in = {}) = 0;
 
-  virtual sharedConstTensors backwarding_with_val(int iteration, 
-                                         sharedConstTensors deriv,
-                                         sharedConstTensors in = {});
+  virtual sharedConstTensors backwarding_with_val(int iteration,
+                                                  sharedConstTensors deriv,
+                                                  sharedConstTensors in = {});
 
   /**
    * @brief     read layer Weight & Bias data from file
@@ -315,12 +315,12 @@ public:
 
   std::vector<Tensor> getGradient();
 
-  void resizeNetInput(unsigned int size){net_input.resize(size);}
-  
-  void resizeNetOutput(unsigned int size){net_hidden.resize(size);}
+  void resizeNetInput(unsigned int size) { net_input.resize(size); }
 
-  unsigned int getNumInputs(){return num_inputs;}
-  unsigned int getNumOutputs(){return num_outputs;}
+  void resizeNetOutput(unsigned int size) { net_hidden.resize(size); }
+
+  unsigned int getNumInputs() { return num_inputs; }
+  unsigned int getNumOutputs() { return num_outputs; }
 
   void setInputBuffer(unsigned int i, std::shared_ptr<NetBuffers> n_buffer) {
     if (i >= net_input.size())