From 6320858c770c1592ca463f4bf4e5ac45e3a81698 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:29:53 +0400 Subject: [PATCH] Add Softmax operation description (#294) This class is used to represent softmax operation in computation graph Signed-off-by: Vladimir Plazun --- .../nnc/core/IR/model/operations/softmax_op.h | 37 ++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 contrib/nnc/libs/core/include/nnc/core/IR/model/operations/softmax_op.h diff --git a/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/softmax_op.h b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/softmax_op.h new file mode 100644 index 0000000..e9d6f5b --- /dev/null +++ b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/softmax_op.h @@ -0,0 +1,37 @@ +#ifndef _NNC_CORE_IR_MODEL_SOFTMAX_H_ +#define _NNC_CORE_IR_MODEL_SOFTMAX_H_ + +#include "nnc/core/IR/model/operations/operation.h" + +namespace nncc +{ +namespace contrib +{ +namespace core +{ +namespace IR +{ +namespace model +{ +namespace ops +{ + +class SoftmaxOp : public OpDescription +{ +public: + explicit SoftmaxOp(int axis) : OpDescription(1, 1), _axis(axis) {} + + int getAxis() const { return _axis; } + +private: + int _axis; +}; + +} // namespace ops +} // namespace model +} // namespace IR +} // namespace core +} // namespace contrib +} // namespace nncc + +#endif //_NNC_CORE_IR_MODEL_SOFTMAX_H_ -- 2.7.4