SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
smv_sigmoid_op.h
1 #ifndef _OPERATORS_SMV_SMV_SIGMOID_OP_H_
2 #define _OPERATORS_SMV_SMV_SIGMOID_OP_H_
3 
4 #include "smaug/core/backend.h"
6 #include "smaug/operators/sigmoid_op.h"
7 #include "smaug/operators/smv/smv_unary_op_common.h"
8 
9 namespace smaug {
10 
12 class SmvSigmoidOp : public SigmoidOp<SmvBackend> {
13  public:
15  void tile() override { tiledTensors = smv::unary::doTiling(this, false); }
16  void run() override { smv::unary::run(this, tiledTensors); }
17 
18  protected:
19  std::array<TiledTensor, 2> tiledTensors;
20 };
21 
22 } // namespace smaug
23 
24 #endif
smaug::SigmoidOp
Implements the sigmoid operator, defined as 1/(1 + exp(-input)).
Definition: backend.h:56
smaug::SmvSigmoidOp
Sigmoid linear-unit operator on SMV.
Definition: smv_sigmoid_op.h:12
smaug
The smaug namespace is the parent namespace of all C++ code in SMAUG.
Definition: backend.cpp:38
common.h
Utilities for writing and invoking Aladdin kernels from Operators.