|
SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
All operators supported by SMAUG. More...
Classes | |
| class | smaug::BatchNormOp< Backend > |
| Implements the batch normalization layer. More... | |
| class | smaug::ConcatOp< Backend > |
| Concatenates N Tensors along a specified axis. More... | |
| class | smaug::SwitchOp< Backend > |
| Conditionally forwards an input to one of two outputs. More... | |
| class | smaug::MergeOp< Backend > |
| Forwards the first live input to its output. More... | |
| class | smaug::ConvolutionOp< Backend > |
| The base class for all 4D spatial convolution operators. More... | |
| class | smaug::DataOp< Backend > |
| Exposes a Tensor as its only output. More... | |
| class | smaug::DepthwiseConvolutionOp< Backend > |
| Implements the depthwise convolution operator. More... | |
| class | smaug::EltwiseAddOp< Backend > |
| Adds two Tensors elementwise. More... | |
| class | smaug::EltwiseMulOp< Backend > |
| Multiplies two Tensors elementwise. More... | |
| class | smaug::EltwiseOp< Backend > |
| The base class of all elementwise operators. More... | |
| class | smaug::EluOp< Backend > |
| Implements the exponential linear unit function. More... | |
| class | smaug::SeluOp< Backend > |
| Implements the scaled exponential linear unit function. More... | |
| class | smaug::FusedActivationOp |
| An Operator fused with an activation function. More... | |
| class | smaug::GreaterOp< Backend > |
| Implements an elementwise greater than operator. More... | |
| class | smaug::GreaterEqualOp< Backend > |
| Implements an elementwise greater than or equal to operator. More... | |
| class | smaug::InnerProductOp< Backend > |
| Implements the inner product operator. More... | |
| class | smaug::LessOp< Backend > |
| Implements an elementwise less-than operator. More... | |
| class | smaug::LessEqualOp< Backend > |
| Implements an elementwise less-than-or-equal-to operator. More... | |
| class | smaug::PaddingOp< Backend > |
| Pad a given tensor in any number of dimensions with arbitrary size. More... | |
| class | smaug::PoolingOp< Backend > |
| Implements a pooling operator. More... | |
| class | smaug::MaxPoolingOp< Backend > |
| Implements the max-pooling operator. More... | |
| class | smaug::AvgPoolingOp< Backend > |
| Implements the arithmetic-average-pooling operator. More... | |
| class | smaug::ReluOp< Backend > |
| Implements the rectified linear unit operator: max(slope * x, 0). More... | |
| class | smaug::ReorderOp< Backend > |
| Implements a Tensor reordering operation to convert between different DataLayouts. More... | |
| class | smaug::RepeatOp< Backend > |
| Replicates a Tensor's data among all dimensions. More... | |
| class | smaug::ReshapeOp< Backend > |
| Changes the Tensor's shape while retaining the number of elements. More... | |
| class | smaug::SigmoidOp< Backend > |
| Implements the sigmoid operator, defined as 1/(1 + exp(-input)). More... | |
| class | smaug::SoftmaxOp< Backend > |
| Implements the softmax operator. More... | |
| class | smaug::SplitOp< Backend > |
| Implements the split operator, which divides a Tensor into N output Tensors along a specified dimension. More... | |
| class | smaug::TanhOp< Backend > |
| Implements the tanh operator. More... | |
| class | smaug::HardTanhOp< Backend > |
| Implements the hard tanh operator, which bounds the min and max value of the tanh operator. More... | |
| class | smaug::UnaryOp< Backend > |
| Base class for all operators with one input. More... | |
All operators supported by SMAUG.
1.8.18