37 : _fill_func(), _unpooling_layer_kernel()
44 _fill_func = std::make_unique<NEFill>();
45 _unpooling_layer_kernel = std::make_unique<NEMaxUnpoolingLayerKernel>();
46 _fill_func->configure(output, zero_value);
47 _unpooling_layer_kernel->configure(input, indices, output, pool_info);
Class describing the value of a pixel for any image format.
NEMaxUnpoolingLayer()
Constructor.
static Status validate(const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, const PoolingLayerInfo &pool_info)
Static function to check if given info will lead to a valid configuration of NEMaxUnpoolingLayer.
void configure(ITensor *input, ITensor *indices, ITensor *output, const PoolingLayerInfo &pool_info)
Set the input and output tensors.
Store the tensor's metadata.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
static Status validate(const ITensorInfo *input, const ITensorInfo *indices, const ITensorInfo *output, const PoolingLayerInfo &pool_info)
Static function to check if given info will lead to a valid configuration of NEMaxUnpoolingLayerKerne...
Pooling Layer Information struct.
static constexpr size_t DimY
Alias for dimension 1 also known as Y dimension.
virtual void schedule(ICPPKernel *kernel, const Hints &hints)=0
Runs the kernel in the same thread as the caller synchronously.
void run() override
Run the kernels contained in the function.
~NEMaxUnpoolingLayer()
Default destructor.
static IScheduler & get()
Access the scheduler singleton.