commit cf00c7e3f53cb0df648ec1ab62773da063380034
author Jeremy Johnson <jeremy.johnson@arm.com> Thu Jan 04 15:34:20 2024 +0000
committer Eric Kunze <eric.kunze@arm.com> Mon Jan 08 21:40:52 2024 +0000
tree 5e53f339a7858b348a78bfb5097a67d9d887835b
parent bd801960c958db85ae4092d1350ffbd383c3f77c

Main compliance: Fix CONV2D weight generation

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I7b0b14475fc4397baa404d57e03d6ca2f73ea746

reference_model/src/generate/generate_dot_product.cc

diff --git a/reference_model/src/generate/generate_dot_product.cc b/reference_model/src/generate/generate_dot_product.cc
index 130b41d..67190c6 100644
--- a/reference_model/src/generate/generate_dot_product.cc
+++ b/reference_model/src/generate/generate_dot_product.cc
@@ -1,4 +1,4 @@
-// Copyright (c) 2023, ARM Limited.
+// Copyright (c) 2023-2024, ARM Limited.
 //
 //    Licensed under the Apache License, Version 2.0 (the "License");
 //    you may not use this file except in compliance with the License.
@@ -151,7 +151,7 @@
         uint32_t ic = t % IC;
         uint32_t kx = (t / IC) % KW;
         uint32_t ky = ((t / IC) / KW) % KH;
-        uint32_t k  = (ky + KW * kx) * IC + ic;
+        uint32_t k  = (ky * KW + kx) * IC + ic;
 
         data[t] = static_cast<DataType>(generator(k));
     }