Skip to content

Commit adb2f53

Browse files
wr0112358AnthonyBarbier
authored andcommitted
Bugfix: must zero all values in all slices (#343)
1 parent 9168acc commit adb2f53

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/runtime/NEON/functions/NEDeconvolutionLayer.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ void NEDeconvolutionLayer::run()
9292
const int stride_x = _info.stride().first;
9393
const int stride_y = _info.stride().second;
9494

95-
std::fill_n(reinterpret_cast<float *>(_scaled_output.buffer()), _scaled_output.info()->tensor_shape().total_size(), 0.f);
95+
std::fill_n(_scaled_output.buffer(), _scaled_output.info()->total_size(), 0);
9696

9797
// scaled_output is the input for the forward convolution. We copy the input elements to scaled_output
9898
// and insert rows and columns with zeroes depending on the stride values.

0 commit comments

Comments
 (0)