SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
eltwise_add_op.h
1 #ifndef _OPERATORS_ELTWISE_ADD_OP_H_
2 #define _OPERATORS_ELTWISE_ADD_OP_H_
3 
4 #include "smaug/core/backend.h"
5 #include "smaug/core/operator.h"
7 #include "smaug/core/workspace.h"
8 #include "smaug/operators/eltwise_op.h"
9 
10 namespace smaug {
11 
18 template <typename Backend>
19 class EltwiseAddOp : public EltwiseOp<Backend> {
20  public:
21  EltwiseAddOp(const std::string& name, Workspace* workspace)
22  : EltwiseOp<Backend>(name, OpType::EltwiseAdd, workspace) {}
23 
24  void run() override {}
25 };
26 
27 REGISTER_SPECIAL_OP(EltwiseAddOp, ReferenceBackend);
28 
29 } // namespace smaug
30 
31 #endif
tensor_utils.h
Utility functions for copying/printing/tiling tensors.
smaug
The smaug namespace is the parent namespace of all C++ code in SMAUG.
Definition: backend.cpp:38