[nnc core] Add reshape operation description (#392)

This commit adds missing reshape operation description

Signed-off-by: Vladimir Plazun <v.plazun@partner.samsung.com>

diff --git a/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/reshape_op.h b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/reshape_op.h
new file mode 100644 (file)
index 0000000..bf65492
--- /dev/null
+++ b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/reshape_op.h
@@ -0,0 +1,29 @@
+#pragma once
+
+#include "nnc/core/IR/model/operations/operation.h"
+
+namespace nncc
+{
+namespace contrib
+{
+namespace core
+{
+namespace IR
+{
+namespace model
+{
+namespace ops
+{
+
+class ReshapeOp : public OpDescription
+{
+public:
+  explicit ReshapeOp() : OpDescription(1, 1) {}
+};
+
+} // namespace ops
+} // namespace model
+} // namespace IR
+} // namespace core
+} // namespace contrib
+} // namespace nncc

diff --git a/contrib/nnc/libs/core/include/nnc/core/IR/model/visitor/visitor.h b/contrib/nnc/libs/core/include/nnc/core/IR/model/visitor/visitor.h
index 0d16e83..189cd3c 100644 (file)
--- a/contrib/nnc/libs/core/include/nnc/core/IR/model/visitor/visitor.h
+++ b/contrib/nnc/libs/core/include/nnc/core/IR/model/visitor/visitor.h
@@ -25,6 +25,7 @@ namespace ops
   class BiasAddOp;
   class VariableOp;
   class ReluOp;
+  class ReshapeOp;
 }
 
 class Visitor {
@@ -39,6 +40,7 @@ class Visitor {
   virtual void visit(ADT::INode *node, ops::BiasAddOp &op) = 0;
   virtual void visit(ADT::INode *node, ops::VariableOp &op) = 0;
   virtual void visit(ADT::INode *node, ops::ReluOp &op) = 0;
+  virtual void visit(ADT::INode* node, ops::ReshapeOp &op) = 0;
 
   virtual ~Visitor() = default;
 };