Compute Library
18.05
|
Interface for the ROI pooling layer kernel. More...
#include <NEROIPoolingLayerKernel.h>
Public Member Functions | |
const char * | name () const override |
Name of the kernel. More... | |
NEROIPoolingLayerKernel () | |
Default constructor. More... | |
NEROIPoolingLayerKernel (const NEROIPoolingLayerKernel &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEROIPoolingLayerKernel & | operator= (const NEROIPoolingLayerKernel &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEROIPoolingLayerKernel (NEROIPoolingLayerKernel &&)=default | |
Allow instances of this class to be moved. More... | |
NEROIPoolingLayerKernel & | operator= (NEROIPoolingLayerKernel &&)=default |
Allow instances of this class to be moved. More... | |
~NEROIPoolingLayerKernel ()=default | |
Default destructor. More... | |
void | configure (const ITensor *input, const IROIArray *rois, ITensor *output, const ROIPoolingLayerInfo &pool_info) |
Set the input and output tensors. More... | |
void | run (const Window &window, const ThreadInfo &info) override |
Execute the kernel on the passed window. More... | |
Public Member Functions inherited from ICPPKernel | |
virtual | ~ICPPKernel ()=default |
Default destructor. More... | |
Public Member Functions inherited from IKernel | |
IKernel () | |
Constructor. More... | |
virtual | ~IKernel ()=default |
Destructor. More... | |
virtual bool | is_parallelisable () const |
Indicates whether or not the kernel is parallelisable. More... | |
virtual BorderSize | border_size () const |
The size of the border for that kernel. More... | |
const Window & | window () const |
The maximum window the kernel can be executed on. More... | |
Interface for the ROI pooling layer kernel.
Definition at line 36 of file NEROIPoolingLayerKernel.h.
Default constructor.
Referenced by NEROIPoolingLayerKernel::name().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Allow instances of this class to be moved.
|
default |
Default destructor.
Referenced by NEROIPoolingLayerKernel::name().
void configure | ( | const ITensor * | input, |
const IROIArray * | rois, | ||
ITensor * | output, | ||
const ROIPoolingLayerInfo & | pool_info | ||
) |
Set the input and output tensors.
[in] | input | Source tensor. Data types supported: F32. |
[in] | rois | Array containing ROI. |
[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. Referenced by NEROIPoolingLayerKernel::name().
|
inlineoverridevirtual |
Name of the kernel.
Implements ICPPKernel.
Definition at line 39 of file NEROIPoolingLayerKernel.h.
References NEROIPoolingLayerKernel::configure(), arm_compute::test::validation::info(), NEROIPoolingLayerKernel::NEROIPoolingLayerKernel(), NEROIPoolingLayerKernel::operator=(), NEROIPoolingLayerKernel::run(), IKernel::window(), and NEROIPoolingLayerKernel::~NEROIPoolingLayerKernel().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
Referenced by NEROIPoolingLayerKernel::name().
|
default |
Allow instances of this class to be moved.
|
overridevirtual |
Execute the kernel on the passed window.
[in] | window | Region on which to execute the kernel. (Must be a region of the window returned by window()) |
[in] | info | Info about executing thread and CPU. |
Implements ICPPKernel.
Referenced by NEROIPoolingLayerKernel::name().