SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
smaug
operators
smv
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"
5
#include "
smaug/operators/common.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
:
14
using
SigmoidOp<SmvBackend>::SigmoidOp
;
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.
Generated by
1.8.18