ARM Compute Library
17.04
|
Basic function to simulate a pooling layer with the specified pooling operation. More...
#include <NEPoolingLayer.h>
Public Member Functions | |
void | configure (ITensor *input, ITensor *output, const PoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
![]() | |
INESimpleFunction () | |
Constructor. More... | |
void | run () overridefinal |
Run the kernels contained in the function. More... | |
![]() | |
virtual | ~IFunction ()=default |
Destructor. More... | |
Basic function to simulate a pooling layer with the specified pooling operation.
This function calls the following NEON kernels:
Definition at line 40 of file NEPoolingLayer.h.
void configure | ( | ITensor * | input, |
ITensor * | output, | ||
const PoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
[in,out] | input | Source tensor. (Written to only when padding != 0) Data types supported: F32. |
[out] | output | Destination tensor. Data types supported: Same as input . |
[in] | pool_info | Contains pooling operation information described in PoolingLayerInfo. |