SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
eltwise_mul_op.h
1 #ifndef _OPERATORS_ELTWISE_MUL_OP_H_
2 #define _OPERATORS_ELTWISE_MUL_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 EltwiseMulOp : public EltwiseOp<Backend> {
20  public:
21  EltwiseMulOp(const std::string& name, Workspace* workspace)
22  : EltwiseOp<Backend>(name, OpType::EltwiseMul, workspace) {}
23 
24  void run() override {}
25 };
26 
27 REGISTER_SPECIAL_OP(EltwiseMulOp, 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