33 : _kernel(nullptr), _border_handler(), _shift_handler()
40 auto k = std::make_unique<GCPoolingLayerKernel>();
41 k->configure(input, output, pool_info, indices);
42 _kernel = std::move(k);
BorderMode
Methods available to handle borders.
Class describing the value of a pixel for any image format.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const PoolingLayerInfo &pool_info, const ITensorInfo *indices=nullptr)
Static function to check if given info will lead to a valid configuration of GCPoolingLayerKernel.
void dispatch(IGCKernel &kernel, bool flush=true)
Schedule the execution of the passed kernel if possible.
void memory_barrier()
Defines a barrier ordering memory transactions.
Store the tensor's metadata.
Interface for GLES Compute tensor.
void run() override final
Run the kernels contained in the function.
Copyright (c) 2017-2021 Arm Limited.
static GCScheduler & get()
Access the scheduler singleton.
Pooling Layer Information struct.
void configure(const IGCTensor *tensor, BorderSize border_size, BorderMode border_mode, const PixelValue &constant_border_value=PixelValue())
Initialise the kernel's input, output and border mode.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const PoolingLayerInfo &pool_info, const ITensorInfo *indices=nullptr)
Static function to check if given info will lead to a valid configuration of GCPoolingLayer.
void configure(IGCTensor *input, IGCTensor *output, const PoolingLayerInfo &pool_info, IGCTensor *indices=nullptr)
Set the input and output tensors.
Pixels outside the image are assumed to have the same value as the closest image pixel.
Interface to enqueue GLES kernels and get/set the GLES CommandQueue.
void configure(IGCTensor *input)
Set the input of the kernel.