From cf00c7e3f53cb0df648ec1ab62773da063380034 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Thu, 4 Jan 2024 15:34:20 +0000 Subject: Main compliance: Fix CONV2D weight generation Signed-off-by: Jeremy Johnson Change-Id: I7b0b14475fc4397baa404d57e03d6ca2f73ea746 --- reference_model/src/generate/generate_dot_product.cc | 4 ++-- 1 file 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(generator(k)); } -- cgit v1.2.1