SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
smv_greater_op.h
1 #ifndef _OPERATORS_SMV_SMV_GREATER_OP_H_
2 #define _OPERATORS_SMV_SMV_GREATER_OP_H_
3 
4 #include "smaug/core/backend.h"
5 #include "smaug/operators/common.h"
6 #include "smaug/operators/greater_op.h"
7 
8 namespace smaug {
9 
11 class SmvGreaterOp : public GreaterOp<SmvBackend> {
12  public:
13  using GreaterOp<SmvBackend>::GreaterOp;
14  void tile() override;
15  void run() override;
16 
17  protected:
18  void runX(TiledTensor& inputs0, TiledTensor& inputs1, TiledTensor& outputs);
19 
20  std::array<TiledTensor, 3> tiledTensors;
21 };
22 
24 class SmvGreaterEqualOp : public GreaterEqualOp<SmvBackend> {
25  public:
26  using GreaterEqualOp<SmvBackend>::GreaterEqualOp;
27  void tile() override;
28  void run() override;
29 
30  protected:
31  void runX(TiledTensor& inputs0, TiledTensor& inputs1, TiledTensor& outputs);
32 
33  std::array<TiledTensor, 3> tiledTensors;
34 };
35 
36 
37 } // namespace smaug
38 
39 #endif
smaug::TiledTensor
A multidimensional container of Tensors.
Definition: tensor.h:552
smaug::SmvGreaterEqualOp
Elementwise greater-than-or-equal-to operator on SMV.
Definition: smv_greater_op.h:24
smaug::GreaterOp
Implements an elementwise greater than operator.
Definition: backend.h:51
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.
smaug::GreaterEqualOp
Implements an elementwise greater than or equal to operator.
Definition: backend.h:52
smaug::SmvGreaterOp
Elementwise greater-than operator on SMV.
Definition: smv_greater_op.h:11
Generated by   doxygen 1.8.18