COMPMID-2238: Implement SPACE_TO_DEPTH for CL
Change-Id: I2f971d6902a64ee40820b6959e9b0b4224da6947
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1283
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp
index 9806796..4c9808d 100644
--- a/src/core/CL/CLKernelLibrary.cpp
+++ b/src/core/CL/CLKernelLibrary.cpp
@@ -470,6 +470,8 @@
{ "space_to_batch_static_nchw", "space_to_batch.cl" },
{ "space_to_batch_nhwc", "space_to_batch.cl" },
{ "space_to_batch_static_nhwc", "space_to_batch.cl" },
+ { "space_to_depth_nchw", "space_to_depth.cl" },
+ { "space_to_depth_nhwc", "space_to_depth.cl" },
{ "softmax_layer_max_shift_exp_sum_parallel", "softmax_layer.cl" },
{ "stack_layer", "stack_layer.cl" },
{ "strided_slice", "slice_ops.cl" },
@@ -959,6 +961,10 @@
#include "./cl_kernels/space_to_batch.clembed"
},
{
+ "space_to_depth.cl",
+#include "./cl_kernels/space_to_depth.clembed"
+ },
+ {
"stack_layer.cl",
#include "./cl_kernels/stack_layer.clembed"
},