41 _roi_kernel = std::make_unique<NEROIPoolingLayerKernel>();
42 _roi_kernel->configure(input, rois, output, pool_info);
~NEROIPoolingLayer()
Default destructor.
void run() override
Run the kernels contained in the function.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
static constexpr size_t DimX
Alias for dimension 0 also known as X dimension.
void configure(const ITensor *input, const ITensor *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info)
Set the input and output tensors.
NEROIPoolingLayer()
Constructor.
virtual void schedule(ICPPKernel *kernel, const Hints &hints)=0
Runs the kernel in the same thread as the caller synchronously.
ROI Pooling Layer Information class.
static IScheduler & get()
Access the scheduler singleton.