diff options
author | giuros01 <giuseppe.rossini@arm.com> | 2018-10-03 12:44:35 +0100 |
---|---|---|
committer | Anthony Barbier <Anthony.barbier@arm.com> | 2018-11-15 09:13:14 +0000 |
commit | cd96a26f67bfbb9b0efe6e0e2b229d0b46b4e3e6 (patch) | |
tree | be95114ff6f5a8a17dace8734be6239312631cea /src/core/CL/cl_kernels/bounding_box_transform.cl | |
parent | 1c32bf396eb690a54fd94487e3f258b2c7d31753 (diff) | |
download | ComputeLibrary-cd96a26f67bfbb9b0efe6e0e2b229d0b46b4e3e6.tar.gz |
COMPMID-1329: Add support for GenerateProposals operator in CL
Change-Id: Ib0798cc17496b7817f5b5769b25d98913a33a69d
Diffstat (limited to 'src/core/CL/cl_kernels/bounding_box_transform.cl')
-rw-r--r-- | src/core/CL/cl_kernels/bounding_box_transform.cl | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/core/CL/cl_kernels/bounding_box_transform.cl b/src/core/CL/cl_kernels/bounding_box_transform.cl index d33018847e..14a0fadc2f 100644 --- a/src/core/CL/cl_kernels/bounding_box_transform.cl +++ b/src/core/CL/cl_kernels/bounding_box_transform.cl @@ -28,11 +28,11 @@ /** Perform a padded copy of input tensor to the output tensor. Padding values are defined at compile time * * @attention The following variables must be passed at compile time: - * -# -DDATA_TYPE = Tensor data type. Supported data types: F16/F32 + * -# -DDATA_TYPE= Tensor data type. Supported data types: F16/F32 * -# -DWEIGHT{X,Y,W,H}= Weights [wx, wy, ww, wh] for the deltas * -# -DIMG_WIDTH= Original image width * -# -DIMG_HEIGHT= Original image height - * -# -DBOX_FIELDS=Number of fields that are used to represent a box in boxes + * -# -DBOX_FIELDS= Number of fields that are used to represent a box in boxes * * @param[in] boxes_ptr Pointer to the boxes tensor. Supported data types: F16/F32 * @param[in] boxes_stride_x Stride of the boxes tensor in X dimension (in bytes) @@ -97,7 +97,7 @@ __kernel void bounding_box_transform( // Useful vector constant definitions const VEC_DATA_TYPE(DATA_TYPE, 4) - max_values = (VEC_DATA_TYPE(DATA_TYPE, 4))(IMG_WIDTH, IMG_HEIGHT, IMG_WIDTH, IMG_HEIGHT); + max_values = (VEC_DATA_TYPE(DATA_TYPE, 4))(IMG_WIDTH - 1, IMG_HEIGHT - 1, IMG_WIDTH - 1, IMG_HEIGHT - 1); const VEC_DATA_TYPE(DATA_TYPE, 4) sign = (VEC_DATA_TYPE(DATA_TYPE, 4))(-1, -1, 1, 1); const VEC_DATA_TYPE(DATA_TYPE, 4) |