ARM Compute Library
17.04
|
Basic function to simulate a pooling layer with the specified pooling operation. More...
#include <CLPoolingLayer.h>
Public Member Functions | |
void | configure (ICLTensor *input, ICLTensor *output, const PoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
Public Member Functions inherited from ICLSimpleFunction | |
ICLSimpleFunction () | |
Default 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 simulate a pooling layer with the specified pooling operation.
This function calls the following OpenCL kernels:
Definition at line 40 of file CLPoolingLayer.h.
void configure | ( | ICLTensor * | input, |
ICLTensor * | 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: F16, F32. |
[out] | output | Destination tensor. Data types supported: Same as input . |
[in] | pool_info | Contains pooling operation information described in PoolingLayerInfo. |