1 #ifndef _OPERATORS_SIGMOID_OP_H_
2 #define _OPERATORS_SIGMOID_OP_H_
6 #include "smaug/core/backend.h"
7 #include "smaug/operators/unary_op.h"
17 template <
typename Backend>
18 class SigmoidOp :
public UnaryOp<Backend> {
20 SigmoidOp(
const std::string& name, Workspace* workspace)
21 : UnaryOp<Backend>(name, OpType::Sigmoid, workspace) {}
23 void run()
override {}
26 REGISTER_SPECIAL_OP(SigmoidOp, ReferenceBackend);