[AppContext] Integrate layer-devel

[platform/core/ml/nntrainer.git] / nntrainer / layers / plugged_layer.h

diff --git a/nntrainer/layers/plugged_layer.h b/nntrainer/layers/plugged_layer.h
index c6a982e..318690e 100644 (file)
--- a/nntrainer/layers/plugged_layer.h
+++ b/nntrainer/layers/plugged_layer.h
@@ -34,7 +34,7 @@ public:
    *
    * @param pluggable LayerPluggable structure from the symbol
    */
-  PluggedLayer(const nntrainer::LayerPluggable *pluggable) :
+  PluggedLayer(const nntrainer::LayerV1Pluggable *pluggable) :
     /// @todo we won't need dynamic pointer cast here after api is fully
     /// implemented
     layerImpl(pluggable->createfunc()),
@@ -248,7 +248,7 @@ private:
   /// @todo: migrate to ml::train::Layer
   // ml::train::Layer *layerImpl;
   nntrainer::LayerV1 *layerImpl;
-  nntrainer::DestroyLayerFunc destroy_func;
+  nntrainer::DestroyLayerV1Func destroy_func;
 };
 } // namespace internal
 } // namespace nntrainer