From adb2f538210329e2cb26876cdb97643fbca1d657 Mon Sep 17 00:00:00 2001 From: wr0112358 Date: Wed, 14 Feb 2018 18:37:51 +0100 Subject: [PATCH] Bugfix: must zero all values in all slices (#343) --- src/runtime/NEON/functions/NEDeconvolutionLayer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp b/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp index 7bce8a6..61ffc77 100644 --- a/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp @@ -92,7 +92,7 @@ void NEDeconvolutionLayer::run() const int stride_x = _info.stride().first; const int stride_y = _info.stride().second; - std::fill_n(reinterpret_cast(_scaled_output.buffer()), _scaled_output.info()->tensor_shape().total_size(), 0.f); + std::fill_n(_scaled_output.buffer(), _scaled_output.info()->total_size(), 0); // scaled_output is the input for the forward convolution. We copy the input elements to scaled_output // and insert rows and columns with zeroes depending on the stride values. -- 2.7.4