Merge pull request #24108 from LaurentBerger:I24044

Solved bug in Reduce layer #24044
pull/24119/head^2
Alexander Smorkalov 1 year ago committed by GitHub
commit 7d59db4ec4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      modules/dnn/src/layers/reduce_layer.cpp

@ -425,7 +425,7 @@ public:
dtype* p_dst = dst.ptr<dtype>(); dtype* p_dst = dst.ptr<dtype>();
size_t main_index = start / last_unreduced_dim; size_t main_index = start / last_unreduced_dim;
size_t loop = start / last_unreduced_dim; size_t loop = start % last_unreduced_dim;
size_t origin = unprojected_steps[main_index] + loop * last_unreduced_step; size_t origin = unprojected_steps[main_index] + loop * last_unreduced_step;
for (int i = start; i < end; ++i) { for (int i = start; i < end; ++i) {
Op accumulator(n_reduce, p_src[origin + projected_steps[0]]); Op accumulator(n_reduce, p_src[origin + projected_steps[0]]);

Loading…
Cancel
Save