24.02.1
|
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, const 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... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *input, const ITensorInfo *rois, const ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) |
Static function to check if given info will lead to a valid configuration of NEROIPoolingLayerKernel. More... | |
Basic function to run NEROIPoolingLayerKernel.
Definition at line 40 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, | ||
const ITensor * | output, | ||
const ROIPoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
Valid data layouts:
Valid data type configurations:
src0 | src1 | dst |
---|---|---|
F32 | U16 | F32 |
QASYMM8 | U16 | QASYMM8 |
[in] | input | Source tensor. Data types supported: QASYMM8/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 48 of file NEROIPoolingLayer.cpp.
References ARM_COMPUTE_LOG_PARAMS, and arm_compute::test::validation::input.
|
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 CPU kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 59 of file NEROIPoolingLayer.cpp.
References Window::DimX, Scheduler::get(), and IScheduler::schedule().
|
static |
Static function to check if given info will lead to a valid configuration of NEROIPoolingLayerKernel.
[in] | input | Source tensor info. Data types supported: QASYMM8/F32. |
[in] | rois | TensorInfo for rois tensor which is a 2D tensor of size [5,N] (where 5 is the number ROIs). Data types supported: U16 |
[in] | output | Destination tensor info. 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 40 of file NEROIPoolingLayer.cpp.
References arm_compute::test::validation::input, and NEROIPoolingLayerKernel::validate().