|
SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Contains common functions for working with unary operators. More...
Functions | |
| std::pair< activation_type, activation_param_t > | getActivationParams (UnaryOp< SmvBackend > *op) |
| Extract activation function parameters from the Operator and stores them in the C-style structs for passing to Aladdin. | |
| void | runX (UnaryOp< SmvBackend > *op, TiledTensor &inputs, TiledTensor &outputs) |
| A generic tile dispatcher for unary operators. More... | |
| std::array< TiledTensor, 2 > | doTiling (UnaryOp< SmvBackend > *op, bool copyData) |
| void | run (UnaryOp< SmvBackend > *op, std::array< TiledTensor, 2 > &tiledTensors) |
Contains common functions for working with unary operators.
| void smaug::smv::unary::runX | ( | UnaryOp< SmvBackend > * | op, |
| TiledTensor & | inputs, | ||
| TiledTensor & | outputs | ||
| ) |
A generic tile dispatcher for unary operators.
"X" indicates that tiles can be scheduled in any order.
Definition at line 53 of file smv_unary_op_common.cpp.
1.8.18