diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-09-11 11:16:47 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | f1addb665ad668dcd34e18c52e4961a7cf5e3886 (patch) | |
| tree | 11c1b606ab26416dd0b39f1526e7eba60394e7c3 /src/core/CL/CLKernelLibrary.cpp | |
| parent | 96f6769de9d9ebe3e631b81aac9a82934a79b0c4 (diff) | |
| download | ComputeLibrary-f1addb665ad668dcd34e18c52e4961a7cf5e3886.tar.gz | |
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 <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/core/CL/CLKernelLibrary.cpp')
| -rw-r--r-- | src/core/CL/CLKernelLibrary.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
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<std::string, std::string> 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" }, |