diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-04 15:34:20 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-01-08 21:40:52 +0000 |
commit | cf00c7e3f53cb0df648ec1ab62773da063380034 (patch) | |
tree | 5e53f339a7858b348a78bfb5097a67d9d887835b | |
parent | bd801960c958db85ae4092d1350ffbd383c3f77c (diff) | |
download | reference_model-cf00c7e3f53cb0df648ec1ab62773da063380034.tar.gz |
Main compliance: Fix CONV2D weight generation
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I7b0b14475fc4397baa404d57e03d6ca2f73ea746
-rw-r--r-- | reference_model/src/generate/generate_dot_product.cc | 4 |
1 files changed, 2 insertions, 2 deletions
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 @@ bool generateConv2DWeight(const TosaReference::GenerateConfig& cfg, 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)); } |