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 <v.plazun@partner.samsung.com>

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 (file)
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_