24.02.1
|
Basic function to run NEROIAlignLayerKernel. More...
#include <NEROIAlignLayer.h>
Public Member Functions | |
void | configure (const ITensor *input, const ITensor *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
Public Member Functions inherited from INESimpleFunctionNoBorder | |
INESimpleFunctionNoBorder (IRuntimeContext *ctx=nullptr) | |
Constructor. More... | |
INESimpleFunctionNoBorder (const INESimpleFunctionNoBorder &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INESimpleFunctionNoBorder (INESimpleFunctionNoBorder &&)=default | |
Default move constructor. More... | |
INESimpleFunctionNoBorder & | operator= (const INESimpleFunctionNoBorder &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INESimpleFunctionNoBorder & | operator= (INESimpleFunctionNoBorder &&)=default |
Default move assignment operator. More... | |
~INESimpleFunctionNoBorder () | |
Default destructor. More... | |
void | run () override final |
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, ITensorInfo *output, const ROIPoolingLayerInfo &pool_info) |
Static function to check if given info will lead to a valid configuration of NEROIAlignLayerKernel. More... | |
Basic function to run NEROIAlignLayerKernel.
Definition at line 36 of file NEROIAlignLayer.h.
void configure | ( | const ITensor * | input, |
const ITensor * | rois, | ||
ITensor * | output, | ||
const ROIPoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
Valid data layouts:
Valid data type configurations:
src0 | src1 | dst |
---|---|---|
F16 | F16 | F16 |
F32 | F32 | F32 |
QASYMM8 | QASYMM16 | QASYMM8 |
QASYMM8_SIGNED | QASYMM16 | QASYMM8_SIGNED |
[in] | input | Source tensor. Data types supported: QASYMM8/QASYMM8_SIGNED/F16/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: QASYMM16 with scale of 0.125 and 0 offset if input is QASYMM8/QASYMM8_SIGNED, otherwise same as input |
[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 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 42 of file NEROIAlignLayer.cpp.
References ARM_COMPUTE_LOG_PARAMS, and arm_compute::test::validation::input.
|
static |
Static function to check if given info will lead to a valid configuration of NEROIAlignLayerKernel.
[in] | input | Source tensor info. Data types supported: QASYMM8/QASYMM8_SIGNED/F16/F32. |
[in] | rois | ROIs tensor info. Data types supported: QASYMM16 with scale of 0.125 and 0 offset if input is QASYMM8/QASYMM8_SIGNED, otherwise same as input |
[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 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 32 of file NEROIAlignLayer.cpp.
References ARM_COMPUTE_RETURN_ON_ERROR, arm_compute::test::validation::input, and NEROIAlignLayerKernel::validate().