From: Vladimir Plazun/AI Tools Lab/Engineer/삼성전자 Date: Fri, 1 Jun 2018 14:11:10 +0000 (+0400) Subject: Add CappedRelu operation description class (#272) X-Git-Tag: nncc_backup~2637 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=4c3b5f6a80d6a46ccf16f5f716de00baa1d0ac6b;p=platform%2Fcore%2Fml%2Fnnfw.git Add CappedRelu operation description class (#272) * Add CappedRelu operation description class This class used to represent capped relu operation in computation graph Signed-off-by: Vladimir Plazun --- diff --git a/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/capped_relu_op.h b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/capped_relu_op.h new file mode 100644 index 0000000..eaf88c3 --- /dev/null +++ b/contrib/nnc/libs/core/include/nnc/core/IR/model/operations/capped_relu_op.h @@ -0,0 +1,33 @@ +#ifndef _NNC_CORE_IR_MODEL_CAPPED_RELU_H_ +#define _NNC_CORE_IR_MODEL_CAPPED_RELU_H_ + +#include "nnc/core/IR/model/operations/operation.h" + +namespace nncc { +namespace contrib { +namespace core { +namespace IR { +namespace model { +namespace ops { + +class CappedReluOp : public OpDescription { + public: + explicit CappedReluOp(float cap) : OpDescription(1, 1), _cap(cap) { + } + + float getCap() const { + return _cap; + } + + private: + float _cap; +}; + +} // namespace ops +} // namespace model +} // namespace IR +} // namespace core +} // namespace contrib +} // namespace nncc + +#endif //_NNC_CORE_IR_MODEL_CAPPED_RELU_H_