24 #ifndef __ARM_COMPUTE_NEPOOLINGLAYERKERNEL_H__ 25 #define __ARM_COMPUTE_NEPOOLINGLAYERKERNEL_H__ 68 template <PoolingType pooling_type>
69 void pooling2(
const Window &window_input,
const Window &window);
75 template <PoolingType pooling_type>
76 void pooling3(
const Window &window_input,
const Window &window);
85 PoolingFunction _func;
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++.
void run(const Window &window) override
Execute the kernel on the passed window.
Interface for NEON tensor.
NEPoolingLayerKernel()
Default constructor.
Interface for the pooling layer kernel.
void configure(const ITensor *input, ITensor *output, const PoolingLayerInfo &pool_info)
Set the input and output tensors.
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.