Add Concat operation description class (#279)

This class is used to represent concatenation operation in computation graph

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

diff --git a/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/concat_op.h b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/concat_op.h
new file mode 100644 (file)
index 0000000..9359b0f
--- /dev/null
+++ b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/concat_op.h
@@ -0,0 +1,37 @@
+#ifndef _NNC_CORE_IR_MODEL_CONCAT_OP_H_
+#define _NNC_CORE_IR_MODEL_CONCAT_OP_H_
+
+#include "nnc/core/IR/model/operations/operation.h"
+
+namespace nncc
+{
+namespace contrib
+{
+namespace core
+{
+namespace IR
+{
+namespace model
+{
+namespace ops
+{
+
+class ConcatOp : public OpDescription
+{
+public:
+  ConcatOp(int num_inputs, unsigned int axis) : OpDescription(num_inputs, 1), _axis(axis) {}
+
+  unsigned int getAxis() const { return _axis; }
+
+private:
+  unsigned int _axis;
+};
+
+} // namespace ops
+} // namespace model
+} // namespace IR
+} // namespace core
+} // namespace contrib
+} // namespace nncc
+
+#endif //_NNC_CORE_IR_MODEL_CONCAT_OP_H_