Support Logistic in base loader (#7942)

- Add logistic operation on base loader

Signed-off-by: Chunseok Lee <chunseok.lee@samsung.com>

diff --git a/runtimes/neurun/frontend/base_loader/base_loader.h b/runtimes/neurun/frontend/base_loader/base_loader.h
index 9f741af..765a045 100644 (file)
--- a/runtimes/neurun/frontend/base_loader/base_loader.h
+++ b/runtimes/neurun/frontend/base_loader/base_loader.h
@@ -105,6 +105,7 @@ protected:
   void loadTranspose(const Operator *op);
   void loadMean(const Operator *op);
   void loadPad(const Operator *op);
+  void loadLogistic(const Operator *op);
   void loadCustom(const Operator *op);
 
 protected:
@@ -657,6 +658,18 @@ void BaseLoader<LoaderDomain, SpecificLoader>::loadPad(const Operator *op)
 }
 
 template <typename LoaderDomain, typename SpecificLoader>
+void BaseLoader<LoaderDomain, SpecificLoader>::loadLogistic(const Operator *op)
+{
+  model::OperandIndexSequence inputs;
+  model::OperandIndexSequence outputs;
+
+  loadOperationIO(op, inputs, outputs);
+
+  std::unique_ptr<model::Operation> new_op(new model::operation::LogisticNode(inputs, outputs));
+  _graph.addOperation(std::move(new_op));
+}
+
+template <typename LoaderDomain, typename SpecificLoader>
 void BaseLoader<LoaderDomain, SpecificLoader>::loadCustom(const Operator *op)
 {
   model::OperandIndexSequence inputs;
@@ -769,6 +782,9 @@ void BaseLoader<LoaderDomain, SpecificLoader>::loadOperation(const Operator *op)
     case BuiltinOperator::BuiltinOperator_PAD:
       loadPad(op);
       return;
+    case BuiltinOperator::BuiltinOperator_LOGISTIC:
+      loadLogistic(op);
+      return;
     case BuiltinOperator::BuiltinOperator_CUSTOM:
       loadCustom(op);
       return;