24 #ifndef __ARM_COMPUTE_NEACTIVATIONLAYERKERNEL_H__ 25 #define __ARM_COMPUTE_NEACTIVATIONLAYERKERNEL_H__ 31 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC 43 const char *
name()
const override 45 return "NEActivationLayerKernel";
92 template <ActivationLayerInfo::ActivationFunction F,
typename T>
93 typename std::enable_if<std::is_same<T, float>::value,
void>::type activation(
const Window &window);
94 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC 99 template <ActivationLayerInfo::ActivationFunction F,
typename T>
100 typename std::enable_if<std::is_same<T, float16_t>::value,
void>::type activation(
const Window &window);
106 template <ActivationLayerInfo::ActivationFunction F,
typename T>
107 typename std::enable_if<std::is_same<T, qint8_t>::value,
void>::type activation(
const Window &window);
112 template <ActivationLayerInfo::ActivationFunction F,
typename T>
113 typename std::enable_if<std::is_same<T, qasymm8_t>::value,
void>::type activation(
const Window &window);
118 template <ActivationLayerInfo::ActivationFunction F,
typename T>
119 typename std::enable_if<std::is_same<T, qint16_t>::value,
void>::type activation(
const Window &window);
124 ActivationFunctionExecutorPtr _func;
NEActivationLayerKernel & operator=(const NEActivationLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Common interface for all kernels implemented in C++.
void configure(ITensor *input, ITensor *output, ActivationLayerInfo activation_info)
Set the input and output tensor.
const char * name() const override
Name of the kernel.
Store the tensor's metadata.
Interface for the activation layer kernel.
Activation Layer Information class.
Interface for NEON tensor.
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
src info() -> set_format(Format::S16)
This file contains all available output stages for GEMMLowp on OpenCL.
ActivationFunction
Available activation functions.
Information about executing thread and CPU.
NEActivationLayerKernel()
Constructor.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info)
Static function to check if given info will lead to a valid configuration of NEActivationLayerKernel...