COMPMID-459 Collapse CL Im2col's higher dimensions
Change-Id: I0ccc39cbcf6926e6810faf3fe264c4af7adc3f7b
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/83070
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/src/core/CL/ICLKernel.cpp b/src/core/CL/ICLKernel.cpp
index 5bd7142..bace631 100644
--- a/src/core/CL/ICLKernel.cpp
+++ b/src/core/CL/ICLKernel.cpp
@@ -122,6 +122,11 @@
add_tensor_argument<3>(idx, tensor, window);
}
+void ICLKernel::add_4D_tensor_argument(unsigned int &idx, const ICLTensor *tensor, const Window &window)
+{
+ add_tensor_argument<4>(idx, tensor, window);
+}
+
unsigned int ICLKernel::num_arguments_per_1D_tensor() const
{
return num_arguments_per_tensor<1>();
@@ -137,6 +142,11 @@
return num_arguments_per_tensor<3>();
}
+unsigned int ICLKernel::num_arguments_per_4D_tensor() const
+{
+ return num_arguments_per_tensor<4>();
+}
+
void ICLKernel::set_target(cl::Device &device)
{
_target = get_target_from_device(device);