24 #ifndef __ARM_COMPUTE_NEPOOLINGLAYERKERNEL_H__ 25 #define __ARM_COMPUTE_NEPOOLINGLAYERKERNEL_H__ 37 const char *
name()
const override 39 return "NEPoolingLayerKernel";
84 template <PoolingType pooling_type,
bool exclude_padding = false>
85 void pooling2_f32_nchw(
const Window &window_input,
const Window &window);
91 template <PoolingType pooling_type,
bool exclude_padding = false>
92 void pooling2_f16_nchw(
const Window &window_input,
const Window &window);
98 template <PoolingType pooling_type>
99 void pooling2_q8_nchw(
const Window &window_input,
const Window &window);
105 template <PoolingType pooling_type,
bool exclude_padding = false>
106 void pooling2_qasymm8_nchw(
const Window &window_input,
const Window &window);
112 template <PoolingType pooling_type>
113 void pooling2_q16_nchw(
const Window &window_input,
const Window &window);
119 template <PoolingType pooling_type,
bool exclude_padding = false>
120 void pooling3_f32_nchw(
const Window &window_input,
const Window &window);
126 template <PoolingType pooling_type,
bool exclude_padding = false>
127 void pooling3_f16_nchw(
const Window &window_input,
const Window &window);
133 template <PoolingType pooling_type>
134 void pooling3_q8_nchw(
const Window &window_input,
const Window &window);
140 template <PoolingType pooling_type,
bool exclude_padding = false>
141 void pooling3_qasymm8_nchw(
const Window &window_input,
const Window &window);
147 template <PoolingType pooling_type>
148 void pooling3_q16_nchw(
const Window &window_input,
const Window &window);
154 template <PoolingType pooling_type,
bool exclude_padding = false>
155 void pooling7_f32_nchw(
const Window &window_input,
const Window &window);
161 template <PoolingType pooling_type>
162 void poolingMxN_q8_nchw(
const Window &window_input,
const Window &window);
168 template <PoolingType pooling_type,
bool exclude_padding = false>
169 void poolingMxN_qasymm8_nchw(
const Window &window_input,
const Window &window);
175 template <PoolingType pooling_type,
bool exclude_padding = false>
176 void poolingMxN_qasymm8_nhwc(
const Window &window_input,
const Window &window);
182 template <PoolingType pooling_type>
183 void poolingMxN_q16_nchw(
const Window &window_input,
const Window &window);
189 template <PoolingType pooling_type,
bool exclude_padding = false>
190 void poolingMxN_f16_nchw(
const Window &window_input,
const Window &window);
196 template <PoolingType pooling_type,
bool exclude_padding = false>
197 void poolingMxN_f16_nhwc(
const Window &window_input,
const Window &window);
203 template <PoolingType pooling_type,
bool exclude_padding = false>
204 void poolingMxN_f32_nchw(
const Window &window_input,
const Window &window);
210 template <PoolingType pooling_type,
bool exclude_padding = false>
211 void poolingMxN_f32_nhwc(
const Window &window_input,
const Window &window);
220 PoolingFunction _func;
224 unsigned int _num_elems_processed_per_iteration;
BorderSize border_size() const override
The size of the border for that kernel.
~NEPoolingLayerKernel()=default
Default destructor.
Container for 2D border size.
Common interface for all kernels implemented in C++.
Store the tensor's metadata.
Interface for NEON tensor.
src info() -> set_format(Format::S16)
This file contains all available output stages for GEMMLowp on OpenCL.
NEPoolingLayerKernel()
Default constructor.
const char * name() const override
Name of the kernel.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const PoolingLayerInfo &pool_info)
Static function to check if given info will lead to a valid configuration of NEPoolingLayerKernel.
Interface for the pooling layer kernel.
void configure(const ITensor *input, ITensor *output, const PoolingLayerInfo &pool_info)
Set the input and output tensors.
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
Information about executing thread and CPU.
NEPoolingLayerKernel & operator=(const NEPoolingLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Pooling Layer Information class.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.