24.02.1
|
Go to the source code of this file.
Macros | |
#define | MAX(x, y) fmax(x, y) |
Functions | |
DATA_TYPE | vec4_max (DATA_TYPE4 vec) |
DATA_TYPE | vec8_max (DATA_TYPE8 vec) |
DATA_TYPE | vec16_max (DATA_TYPE16 vec) |
DATA_TYPE | roi_pool_1x1 (const Tensor3D *input, int region_start_x, int region_end_x, int region_start_y, int region_end_y, int pz) |
Performs a roi pooling on a single output pixel. More... | |
__kernel void | roi_pooling_layer (__global uchar *input_ptr, uint input_stride_x, uint input_step_x, uint input_stride_y, uint input_step_y, uint input_stride_z, uint input_step_z, uint input_offset_first_element_in_bytes, __global uchar *rois_ptr, uint rois_stride_x, uint rois_step_x, uint rois_stride_y, uint rois_step_y, uint rois_offset_first_element_in_bytes, __global uchar *output_ptr, uint output_stride_x, uint output_step_x, uint output_stride_y, uint output_step_y, uint output_stride_z, uint output_step_z, uint output_offset_first_element_in_bytes, unsigned int input_stride_w, unsigned int output_stride_w) |
Performs a roi pooling function. More... | |
#define MAX | ( | x, | |
y | |||
) | fmax(x, y) |
Definition at line 44 of file roi_pooling_layer.cl.
|
inline |
Performs a roi pooling on a single output pixel.
[in] | input | Pointer to input Tensor3D struct. |
[in] | region_start_x | Start x index projected onto the input tensor. |
[in] | region_end_x | End x index projected onto the input tensor. |
[in] | region_start_y | Start y index projected onto the input tensor. |
[in] | region_end_y | End y index projected onto the input tensor. |
[in] | pz | z index of the input tensor. |
Definition at line 79 of file roi_pooling_layer.cl.
References arm_compute::test::validation::input, MAX, MIN_VALUE, QUANTIZE, tensor3D_offset(), VEC_DATA_TYPE, VEC_SIZE, and VLOAD.
Referenced by roi_pooling_layer().
__kernel void roi_pooling_layer | ( | __global uchar * | input_ptr, |
uint | input_stride_x, | ||
uint | input_step_x, | ||
uint | input_stride_y, | ||
uint | input_step_y, | ||
uint | input_stride_z, | ||
uint | input_step_z, | ||
uint | input_offset_first_element_in_bytes, | ||
__global uchar * | rois_ptr, | ||
uint | rois_stride_x, | ||
uint | rois_step_x, | ||
uint | rois_stride_y, | ||
uint | rois_step_y, | ||
uint | rois_offset_first_element_in_bytes, | ||
__global uchar * | output_ptr, | ||
uint | output_stride_x, | ||
uint | output_step_x, | ||
uint | output_stride_y, | ||
uint | output_step_y, | ||
uint | output_stride_z, | ||
uint | output_step_z, | ||
uint | output_offset_first_element_in_bytes, | ||
unsigned int | input_stride_w, | ||
unsigned int | output_stride_w | ||
) |
Performs a roi pooling function.
[in] | input_ptr | Pointer to the source image. Supported data types: F16, F32, QASYMM8 |
[in] | input_stride_x | Stride of the source image in X dimension (in bytes) |
[in] | input_step_x | input_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | input_stride_y | Stride of the source image in Y dimension (in bytes) |
[in] | input_step_y | input_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | input_stride_z | Stride of the source tensor in Z dimension (in bytes) |
[in] | input_step_z | input_stride_z * number of elements along Z processed per workitem(in bytes) |
[in] | input_offset_first_element_in_bytes | The offset of the first element in the pooled region of the source image as specifed by ROI |
[in] | rois_ptr | Pointer to the ROIs tensor. Layout: { batch_index, x1, y1, x2, y2 }. Supported data types: same as input_ptr |
[in] | rois_stride_x | Stride of the ROIs tensor in X dimension (in bytes) |
[in] | rois_step_x | Step of the ROIs tensor in X dimension (in bytes) |
[in] | rois_stride_y | Stride of the ROIs tensor in Y dimension (in bytes) |
[in] | rois_step_y | Step of the ROIs tensor in Y dimension (in bytes) |
[in] | rois_offset_first_element_in_bytes | The offset of the first element in the ROIs tensor |
[out] | output_ptr | Pointer to the destination image. Supported data types: same as input |
[in] | output_stride_x | Stride of the destination image in X dimension (in bytes) |
[in] | output_step_x | output_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | output_stride_y | Stride of the destination image in Y dimension (in bytes) |
[in] | output_step_y | output_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | output_stride_z | Stride of the destination tensor in Z dimension (in bytes) |
[in] | output_step_z | output_stride_z * number of elements along Z processed per workitem(in bytes) |
[in] | output_offset_first_element_in_bytes | The offset of the first element in the destination image |
[in] | input_stride_w | Stride of the source image in W dimension (in bytes) |
[in] | output_stride_w | Stride of the destination image in W dimension (in bytes) |
Definition at line 151 of file roi_pooling_layer.cl.
References arm_compute::utility::clamp(), CONVERT_TO_IMAGE_STRUCT_NO_STEP, CONVERT_TO_TENSOR3D_STRUCT_NO_STEP, arm_compute::test::validation::input, offset(), Tensor3D::ptr, roi_pool_1x1(), arm_compute::round(), tensor3D_offset(), and VEC_DATA_TYPE.
|
inline |
Definition at line 61 of file roi_pooling_layer.cl.
References MAX, vec8_max(), and VEC_DATA_TYPE.
|
inline |
Definition at line 47 of file roi_pooling_layer.cl.
References MAX, and VEC_DATA_TYPE.
Referenced by vec8_max().
|
inline |
Definition at line 54 of file roi_pooling_layer.cl.
References MAX, vec4_max(), and VEC_DATA_TYPE.
Referenced by vec16_max().