diff options
Diffstat (limited to 'tests/validation/reference/DepthwiseConvolutionLayer.cpp')
| -rw-r--r-- | tests/validation/reference/DepthwiseConvolutionLayer.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/validation/reference/DepthwiseConvolutionLayer.cpp b/tests/validation/reference/DepthwiseConvolutionLayer.cpp index 08caa8efb8..6ca347f1d4 100644 --- a/tests/validation/reference/DepthwiseConvolutionLayer.cpp +++ b/tests/validation/reference/DepthwiseConvolutionLayer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -156,17 +156,17 @@ SimpleTensor<uint8_t> depthwise_convolution(const SimpleTensor<uint8_t> &src, co { for(int x = minimum_x; x < input_width + pad_x - filter_half_size; x += conv_info.stride().first) { - Coordinates coords(x, y, z); + Coordinates coords(x, y, z, r); int filter_offset = filter_plane * z; - uint32_t val = 0; + int32_t val = 0; for(int j = y - filter_half_size; j <= (y + filter_half_size); ++j) { for(int i = x - filter_half_size; i <= (x + filter_half_size); ++i) { coords.set(0, i); coords.set(1, j); - auto in_val = tensor_elem_at<uint8_t>(src, coords, BorderMode::CONSTANT, 0); + auto in_val = tensor_elem_at<uint8_t>(src, coords, BorderMode::CONSTANT, -input_offset); uint8_t w_val = *(weights.data() + filter_offset); val += (in_val + input_offset) * (w_val + weights_offset); ++filter_offset; |