ARM Compute Library
17.04
|
Basic function to run NEActivationLayerKernel. More...
#include <NEActivationLayer.h>
Public Member Functions | |
void | configure (const ITensor *input, ITensor *output, ActivationLayerInfo activation_info) |
Set the input and output tensor. More... | |
Public Member Functions inherited from INESimpleFunction | |
INESimpleFunction () | |
Constructor. More... | |
void | run () overridefinal |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from IFunction | |
virtual | ~IFunction ()=default |
Destructor. More... | |
Basic function to run NEActivationLayerKernel.
Definition at line 39 of file NEActivationLayer.h.
void configure | ( | const ITensor * | input, |
ITensor * | output, | ||
ActivationLayerInfo | activation_info | ||
) |
Set the input and output tensor.
[in] | input | Source tensor. Data type supported: F32. |
[out] | output | Destination tensor. Data type supported: same as input |
[in] | activation_info | Activation layer parameters. |