SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
smv_softmax_op.h
1 #ifndef _OPERATORS_SMV_SMV_SOFTMAX_OP_H_
2 #define _OPERATORS_SMV_SMV_SOFTMAX_OP_H_
3 
4 #include "smaug/core/backend.h"
5 #include "smaug/operators/common.h"
6 #include "smaug/operators/softmax_op.h"
7 
8 namespace smaug {
9 
11 class SmvSoftmaxOp : public SoftmaxOp<SmvBackend> {
12  public:
13  using SoftmaxOp<SmvBackend>::SoftmaxOp;
14  void tile() override;
15  void run() override;
16 
17  protected:
18  std::array<TiledTensor, 2> tiledTensors;
19 };
20 
21 } // namespace smaug
22 
23 #endif
smaug::SoftmaxOp
Implements the softmax operator.
Definition: backend.h:39
smaug::SmvSoftmaxOp
Softmax operator on SMV.
Definition: smv_softmax_op.h:11
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   doxygen 1.8.18