From d7ba5397b676c966cb5069c7187a12a0c35305f5 Mon Sep 17 00:00:00 2001
From: Sanghoon Lee <sanghoon.lee@arm.com>
Date: Wed, 13 Dec 2017 11:28:50 +0000
Subject: COMPMID-727 - Implement reference and CL/NEON validation for
 CustomConvolutionRectangle

Change-Id: I108a48ad5e6dc3f331fd5ceb38ced8ccdb31d81a
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/113130
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
---
 tests/validation/reference/Convolution.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'tests/validation/reference/Convolution.cpp')

diff --git a/tests/validation/reference/Convolution.cpp b/tests/validation/reference/Convolution.cpp
index 84be858cfc..777e2df400 100644
--- a/tests/validation/reference/Convolution.cpp
+++ b/tests/validation/reference/Convolution.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2017 ARM Limited.
+ * Copyright (c) 2017, 2018 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -35,7 +35,7 @@ namespace validation
 namespace reference
 {
 template <typename T>
-SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, T constant_border_value, const unsigned int filter_size)
+SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, T constant_border_value, const unsigned int width, const unsigned int height)
 {
     SimpleTensor<T>       dst(src.shape(), src.data_type());
     SimpleTensor<int32_t> sum(src.shape(), src.data_type());
@@ -43,7 +43,7 @@ SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uin
     for(int element_idx = 0; element_idx < src.num_elements(); ++element_idx)
     {
         const Coordinates id = index2coord(src.shape(), element_idx);
-        apply_2d_spatial_filter(id, src, sum, TensorShape(filter_size, filter_size), conv, 1, border_mode, constant_border_value);
+        apply_2d_spatial_filter(id, src, sum, TensorShape(width, height), conv, 1, border_mode, constant_border_value);
 
         if(tensor_elem_at<int32_t>(sum, id, border_mode, constant_border_value) < 0)
         {
@@ -63,7 +63,7 @@ SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uin
 }
 
 template SimpleTensor<uint8_t> convolution(const SimpleTensor<uint8_t> &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, uint8_t constant_border_value,
-                                           const unsigned int filter_size);
+                                           const unsigned int widht, const unsigned int height);
 } // namespace reference
 } // namespace validation
 } // namespace test
-- 
cgit v1.2.1