An Operator fused with an activation function. More...
#include <fused_activation_op.h>
Inheritance diagram for smaug::FusedActivationOp:
Detailed Description
An Operator fused with an activation function.
This is an optimized operator that reduces memory/compute by directly computing the activation function on its output. This is a parent class of all operator implementations that support activation op fusion.
Definition at line 20 of file fused_activation_op.h.
Public Member Functions | |
| FusedActivationOp (const std::string &name, OpType opType, Workspace *workspace) | |
| void | setActivation (ActivationInfo _actInfo) |
| ActivationInfo | getActivation () const |
 Public Member Functions inherited from smaug::Operator | |
| Operator (const std::string &_name, OpType _opType, Workspace *_workspace) | |
| virtual void | tile () |
| virtual void | run ()=0 |
| Executes the Operator. More... | |
| virtual bool | validate () |
| Returns true if the parameters/tensors of this operator are all valid. | |
| virtual void | createAllTensors () |
| For tests: creates all input and output tensors for this operator. More... | |
| virtual bool | isDead () |
| Returns true if the Operator is dead. More... | |
| virtual std::vector< TensorBase * > | getParameterizableInputs () |
| Return a list of Tensors whose values that are parameterizable. More... | |
| virtual int | getNumParameters () const |
| This returns the number of parameterizable weights in the operator. | |
| virtual bool | isSamplingSupported () const |
| virtual void | setSamplingInfo (const SamplingInfo &sampling) |
| void | printSummary (std::ostream &out) const |
| void | setInput (TensorBase *op, int index) |
| void | setOutput (TensorBase *op, int index) |
| void | setNumPendingInputs (int num) |
| Set the number of input tensors that this operator is waiting on. More... | |
| int | getNumPendingInputs () const |
| void | decrNumPendingInputs () |
| const std::string & | getName () const |
| Vertex | getVertex () const |
| void | setVertex (Vertex v) |
| OpType | getOpType () const |
| Workspace * | getWorkspace () |
| Tensor * | getInput (int index) const |
| const std::vector< TensorBase * > & | getInputs () const |
| Tensor * | getOutput (int index) const |
| const std::vector< TensorBase * > & | getOutputs () const |
| void | setInputsMemType (MemoryType type) |
| void | setWeightsMemType (MemoryType type) |
| void | setOutputsMemType (MemoryType type) |
| MemoryType | getInputsMemType () const |
| MemoryType | getWeightsMemType () const |
| MemoryType | getOutputsMemType () const |
Protected Attributes | |
| ActivationInfo | actInfo |
| Protected Attributes inherited from smaug::Operator | |
| std::vector< TensorBase * > | inputs |
| An ordered list of input tensors consumed by this operator. More... | |
| std::vector< TensorBase * > | outputs |
| An ordered list of output tensors produced by this operator. More... | |
| std::string | name |
| OpType | opType |
| Vertex | vertex |
| The BGL Vertex corresponding to this Operator. | |
| Workspace * | workspace |
| int | numPendingInputs |
| The number of tensors that this operator is waiting on before it can be scheduled. | |
| MemoryType | inputsMemType |
| The memory interface over which input activations are expected to arrive. | |
| MemoryType | weightsMemType |
| The memory interface over which weights are expected to arrive. | |
| MemoryType | outputsMemType |
| The memory interface over which outputs are expected to be delivered. | |
The documentation for this class was generated from the following file:
- smaug/operators/fused_activation_op.h
