convert more layer types

Change-Id: Id8ac25ec946923973fcd36c6890fb68af3464021
Signed-off-by: Inki Dae <inki.dae@samsung.com>

diff --git a/src/training_engine_nntrainer.cpp b/src/training_engine_nntrainer.cpp
index 9ba94015f2e27f4f328bd71151f2454e4f9e4a48..1f6b5a58312888f98a1541228f9d029d53e43b80 100644 (file)
--- a/src/training_engine_nntrainer.cpp
+++ b/src/training_engine_nntrainer.cpp
@@ -79,6 +79,10 @@ namespace NntrainerImpl
                                return ML_TRAIN_LAYER_TYPE_CONCAT;
                        case TRAINING_LAYER_TYPE_MULTIOUT:
                                return ML_TRAIN_LAYER_TYPE_MULTIOUT;
+                       case TRAINING_LAYER_TYPE_L2NORM:
+                               return ML_TRAIN_LAYER_TYPE_PREPROCESS_L2NORM;
+                       case TRAINING_LAYER_TYPE_CENTROID_KNN:
+                               return ML_TRAIN_LAYER_TYPE_CENTROID_KNN;
                        default:
                                return ML_TRAIN_LAYER_TYPE_UNKNOWN;
                }
@@ -186,7 +190,7 @@ namespace NntrainerImpl
 
                LOGI("layer property size : %zu", property.options.size());
 
-               for (auto opt : property.options) {
+               for (auto& opt : property.options) {
                        LOGI("Set %s property", opt.c_str());
 
                        int ret = ml_train_layer_set_property(static_cast<ml_train_layer_h>(layer->layer_handle),