From 163d0b27b054e5840ef46c7b9ac3ae0b118b933f Mon Sep 17 00:00:00 2001 From: Sungeun Kim Date: Thu, 27 May 2021 02:21:42 +0900 Subject: [PATCH] [IE CLDNN] Fixed a region_yolo bug (#51761) (#5826) Signed-off-by: Kim, SungEun --- .../kernel_selector/core/cl_kernels/region_yolo_gpu_ref.cl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/region_yolo_gpu_ref.cl b/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/region_yolo_gpu_ref.cl index 366e1d22586..5ef43e53732 100644 --- a/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/region_yolo_gpu_ref.cl +++ b/inference-engine/thirdparty/clDNN/kernel_selector/core/cl_kernels/region_yolo_gpu_ref.cl @@ -10,7 +10,7 @@ inline INPUT0_TYPE FUNC(logistic_activate)(INPUT0_TYPE x) { inline int FUNC(output_index)(int batch, int region_num, int x, int y, int xy, int feature_offset) { #if DO_SOFTMAX - return OUTPUT_GET_INDEX(batch, feature_offset * INPUT0_SIZE_X * INPUT0_SIZE_Y + xy, 1, 1); + return OUTPUT_GET_INDEX(batch, feature_offset * INPUT0_SIZE_X * INPUT0_SIZE_Y + xy, 0, 0); #else return OUTPUT_GET_INDEX(batch, feature_offset, y, x); #endif