relu_op.h

#ifndef _OPERATORS_RELU_OP_H_
#define _OPERATORS_RELU_OP_H_
 
#include <string>
 
#include "smaug/core/backend.h"
#include "smaug/operators/unary_op.h"
 
namespace smaug {
 
template <typename Backend>
class ReluOp : public UnaryOp<Backend> {
   public:
    ReluOp(const std::string& name, Workspace* workspace, float _slope = 0)
            : UnaryOp<Backend>(name, OpType::ReLU, workspace), slope(_slope) {}
 
    void run() override {}
    void setSlope(float _slope) { slope = _slope; }
    float getSlope () const { return slope; }
 
   protected:
    // Slope in the negative region.
    float slope;
};
 
REGISTER_SPECIAL_OP(ReluOp, ReferenceBackend);
 
 
}  // namespace smaug
 
#endif