SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
Functions
smaug::smv::unary Namespace Reference

Contains common functions for working with unary operators. More...

Functions

std::pair< activation_type, activation_param_tgetActivationParams (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)
 

Detailed Description

Contains common functions for working with unary operators.

Function Documentation

◆ runX()

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.