21.02
|
Basic function to run NEROIPoolingLayerKernel. More...
#include <NEROIPoolingLayer.h>
Public Member Functions | |
NEROIPoolingLayer () | |
Constructor. More... | |
NEROIPoolingLayer (const NEROIPoolingLayer &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEROIPoolingLayer & | operator= (const NEROIPoolingLayer &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEROIPoolingLayer (NEROIPoolingLayer &&)=delete | |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
NEROIPoolingLayer & | operator= (NEROIPoolingLayer &&)=delete |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
~NEROIPoolingLayer () | |
Default destructor. More... | |
void | configure (const ITensor *input, const ITensor *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
void | run () override |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from IFunction | |
virtual | ~IFunction ()=default |
Destructor. More... | |
virtual void | prepare () |
Prepare the function for executing. More... | |
Basic function to run NEROIPoolingLayerKernel.
This function calls the following Neon kernels:
Definition at line 44 of file NEROIPoolingLayer.h.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
delete |
Prevent instances of this class from being moved (As this class contains non movable objects)
|
default |
Default destructor.
void configure | ( | const ITensor * | input, |
const ITensor * | rois, | ||
ITensor * | output, | ||
const ROIPoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
[in] | input | Source tensor. Data types supported: F32. |
[in] | rois | ROIs tensor, it is a 2D tensor of size [5, N] (where N is the number of ROIs) containing top left and bottom right corner as coordinate of an image and batch_id of ROI [ batch_id, x1, y1, x2, y2 ]. Data types supported: U16 |
[out] | output | Destination tensor. Data types supported: Same as input . |
[in] | pool_info | Contains pooling operation information described in ROIPoolingLayerInfo. |
output
tensor must be the same as that specified by pool_info
's pooled width and pooled height. output
tensor and input
tensor must be the same. output
tensor must be the same as the number of elements in rois
array. Definition at line 39 of file NEROIPoolingLayer.cpp.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
delete |
Prevent instances of this class from being moved (As this class contains non movable objects)
|
overridevirtual |
Run the kernels contained in the function.
For Neon kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 45 of file NEROIPoolingLayer.cpp.
References Window::DimX, Scheduler::get(), and IScheduler::schedule().