Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / kernel_selector / core / cl_kernels / arg_max_min_gpu_ref.cl
index 3ad4ac6..7db799b 100644 (file)
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-
 #include "include/common.cl"
 #include "include/data_types.cl"
     
 #define GLOBAL_SIZE 128
 #define LOCAL_SIZE GLOBAL_SIZE
 
-typedef struct /* Index and Value type that holds index and value used in this kernel */
-{
-    uint index; 
-    UNIT_TYPE value; 
-} iav_type;
-
 #ifdef MAX_OUT
     #define COMPARE_SIGN <
     #define UNIT_FILL_VAL UNIT_VAL_MIN
@@ -36,6 +29,7 @@ typedef struct /* Index and Value type that holds index and value used in this k
 __attribute__((reqd_work_group_size(LOCAL_SIZE, 1, 1)))
 KERNEL(arg_max_gpu_top_k)(const __global UNIT_TYPE* input, __global float* output)
 {
+#include "include/arg_max_min_common.cl"
     uint results[TOP_K];
     __local iav_type scratch[LOCAL_SIZE];