21.02
|
Basic function to run CLROIAlignLayerKernel. More...
#include <CLROIAlignLayer.h>
Public Member Functions | |
void | configure (const ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
void | configure (const CLCompileContext &compile_context, const ICLTensor *input, const ICLTensor *rois, ICLTensor *output, const ROIPoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
Public Member Functions inherited from ICLSimpleFunction | |
ICLSimpleFunction (CLRuntimeContext *ctx=nullptr) | |
Constructor. More... | |
ICLSimpleFunction (const ICLSimpleFunction &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
ICLSimpleFunction (ICLSimpleFunction &&)=default | |
Default move constructor. More... | |
ICLSimpleFunction & | operator= (const ICLSimpleFunction &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
ICLSimpleFunction & | operator= (ICLSimpleFunction &&)=default |
Default move assignment operator. More... | |
~ICLSimpleFunction () | |
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 CLROIAlignLayer. More... | |
Basic function to run CLROIAlignLayerKernel.
This function calls the following OpenCL kernels:
Definition at line 43 of file CLROIAlignLayer.h.
void configure | ( | const ICLTensor * | input, |
const ICLTensor * | rois, | ||
ICLTensor * | output, | ||
const ROIPoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
[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 39 of file CLROIAlignLayer.cpp.
References CLKernelLibrary::get().
void configure | ( | const CLCompileContext & | compile_context, |
const ICLTensor * | input, | ||
const ICLTensor * | rois, | ||
ICLTensor * | output, | ||
const ROIPoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
[in] | compile_context | The compile context to be used. |
[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 44 of file CLROIAlignLayer.cpp.
|
static |
Static function to check if given info will lead to a valid configuration of CLROIAlignLayer.
[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 CLROIAlignLayer.cpp.
References ARM_COMPUTE_RETURN_ON_ERROR, and CLROIAlignLayerKernel::validate().
Referenced by arm_compute::test::validation::DATA_TEST_CASE().