Merge pull request #24108 from LaurentBerger:I24044

Solved bug in Reduce layer #24044
This commit is contained in:
Alexander Smorkalov 2023-08-07 12:39:05 +03:00 committed by GitHub
commit 7d59db4ec4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -425,7 +425,7 @@ public:
dtype* p_dst = dst.ptr<dtype>();
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;
for (int i = start; i < end; ++i) {
Op accumulator(n_reduce, p_src[origin + projected_steps[0]]);