From f1addb665ad668dcd34e18c52e4961a7cf5e3886 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Tue, 11 Sep 2018 11:16:47 +0100 Subject: COMPMID-1549 Implementing Batch to Space on OpenCL - NHWC Change-Id: If7ae0a8b6255a10711365068d9fb153c71f09818 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/147751 Tested-by: bsgcomp Reviewed-by: Georgios Pinitas --- src/core/CL/CLKernelLibrary.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'src/core/CL/CLKernelLibrary.cpp') diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index fa164542e4..8f5e81bae9 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -153,8 +153,10 @@ const std::map CLKernelLibrary::_kernel_program_map = { "arithmetic_add", "arithmetic_op.cl" }, { "arithmetic_sub", "arithmetic_op.cl" }, { "arithmetic_div", "arithmetic_op.cl" }, - { "batch_to_space", "batch_to_space.cl" }, - { "batch_to_space_static", "batch_to_space.cl" }, + { "batch_to_space_nchw", "batch_to_space.cl" }, + { "batch_to_space_static_nchw", "batch_to_space.cl" }, + { "batch_to_space_nhwc", "batch_to_space.cl" }, + { "batch_to_space_static_nhwc", "batch_to_space.cl" }, { "batchnormalization_layer_nchw", "batchnormalization_layer.cl" }, { "batchnormalization_layer_nhwc", "batchnormalization_layer.cl" }, { "bitwise_or", "bitwise_op.cl" }, -- cgit v1.2.1