add DestroyOptimizer function

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

diff --git a/src/training_engine_nntrainer.cpp b/src/training_engine_nntrainer.cpp
index 7aa82cd9e45e0832b76b962c6f72a46f5353842b..232b738c1de7ab3aa350f1d83f142afadae8d23b 100644 (file)
--- a/src/training_engine_nntrainer.cpp
+++ b/src/training_engine_nntrainer.cpp
@@ -308,6 +308,21 @@ namespace NntrainerImpl
                return optimizer;
        }
 
+       void TrainingNntrainer::DestroyOptimizer(const training_engine_optimizer *optimizer)
+       {
+               LOGI("ENTER");
+
+               TE_CHECK_OBJ_NO_RETURN_VAL(optimizer);
+
+               int ret = ml_train_optimizer_destroy(static_cast<ml_train_optimizer_h>(optimizer->optimizer_handle));
+               if (ret != ML_ERROR_NONE) {
+                       LOGE("Failed to destroy a optimizer handle.");
+                       return;
+               }
+
+               LOGI("LEAVE");
+       }
+
        int TrainingNntrainer::SetOptimizerProperty(const training_engine_optimizer *optimizer,
                                                                                                training_engine_optimizer_property &property)
        {

diff --git a/src/training_engine_nntrainer_private.h b/src/training_engine_nntrainer_private.h
index 406148d883f20375670c92aca5f1fe3c249abd72..5c53f12ad341e0bffb95f9a48f1d2e2065181bd8 100644 (file)
--- a/src/training_engine_nntrainer_private.h
+++ b/src/training_engine_nntrainer_private.h
@@ -59,6 +59,7 @@ namespace NntrainerImpl
                int SetLayerProperty(const training_engine_layer *layer, training_engine_layer_property &property) final;
                int AddLayer(const training_engine_model *model, const training_engine_layer *layer) final;
                std::unique_ptr<training_engine_optimizer> CreateOptimizer(training_engine_optimizer_type_e type) final;
+               void DestroyOptimizer(const training_engine_optimizer *optimizer) final;
                int SetOptimizerProperty(const training_engine_optimizer *optimizer, training_engine_optimizer_property &property) final;
                int AddOptimizer(const training_engine_model *model, const training_engine_optimizer *optimizer) final;
                std::unique_ptr<training_engine_dataset> CreateDataset(void) final;