Interface for the ROI pooling layer kernel.
More...
#include <NEROIPoolingLayerKernel.h>
Interface for the ROI pooling layer kernel.
Definition at line 36 of file NEROIPoolingLayerKernel.h.
Prevent instances of this class from being copied (As this class contains pointers)
Allow instances of this class to be moved.
Set the input and output tensors.
- Parameters
-
[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. |
- Note
- The x and y dimensions of
output
tensor must be the same as that specified by pool_info
's pooled width and pooled height.
-
The z dimensions of
output
tensor and input
tensor must be the same.
-
The fourth dimension of
output
tensor must be the same as the number of elements in rois
array.
Prevent instances of this class from being copied (As this class contains pointers)
Allow instances of this class to be moved.
Execute the kernel on the passed window.
- Warning
- If is_parallelisable() returns false then the passed window must be equal to window()
- Note
- The window has to be a region within the window returned by the window() method
-
The width of the window has to be a multiple of num_elems_processed_per_iteration().
- Parameters
-
[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.
The documentation for this class was generated from the following file: