From 83b8e77e0405ce72f80a38794314ee8967e22b42 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Vladimir=20Plazun/AI=20Tools=20Lab/Engineer/=EC=82=BC?= =?utf8?q?=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 1 Jun 2018 16:23:31 +0400 Subject: [PATCH] Add Concat operation description class (#279) This class is used to represent concatenation operation in computation graph Signed-off-by: Vladimir Plazun --- .../nnc/core/IR/model/operations/concat_op.h | 37 ++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 contrib/nnc/libs/core/include/nnc/core/IR/model/operations/concat_op.h 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 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_ -- 2.7.4