From 16934a511ccd37a28000a9fabb3e6e5fc6f51ec9 Mon Sep 17 00:00:00 2001
From: Michalis Spyrou <michalis.spyrou@arm.com>
Date: Tue, 21 Aug 2018 18:03:58 +0100
Subject: COMPMID-1227 Implementing Space to Batch on OpenCL

Change-Id: I6fd83d6584c56a4fd2470948f1987e23237c16d3
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/145577
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: bsgcomp <bsgcomp@arm.com>
---
 src/core/CL/CLKernelLibrary.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/core/CL/CLKernelLibrary.cpp')

diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp
index 75ff2482c8..ef3a431f1a 100644
--- a/src/core/CL/CLKernelLibrary.cpp
+++ b/src/core/CL/CLKernelLibrary.cpp
@@ -364,6 +364,8 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map =
     { "softmax_layer_max_shift_exp_sum_quantized_serial", "softmax_layer_quantized.cl" },
     { "softmax_layer_max_shift_exp_sum_quantized_parallel", "softmax_layer_quantized.cl" },
     { "softmax_layer_max_shift_exp_sum_serial", "softmax_layer.cl" },
+    { "space_to_batch", "space_to_batch.cl" },
+    { "space_to_batch_static", "space_to_batch.cl" },
     { "softmax_layer_max_shift_exp_sum_parallel", "softmax_layer.cl" },
     { "strided_slice", "slice_ops.cl" },
     { "suppress_non_maximum", "canny.cl" },
@@ -758,6 +760,10 @@ const std::map<std::string, std::string> CLKernelLibrary::_program_source_map =
     {
         "slice_ops.cl",
 #include "./cl_kernels/slice_ops.clembed"
+    },
+    {
+        "space_to_batch.cl",
+#include "./cl_kernels/space_to_batch.clembed"
     },
     {
         "tablelookup.cl",
-- 
cgit v1.2.1