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.