From e942f07388abbdb3b66974a4f41cf9d21e56a4dc Mon Sep 17 00:00:00 2001
From: Eric Kunze <eric.kunze@arm.com>
Date: Fri, 7 Jan 2022 14:27:01 -0800
Subject: Fix range of C iteration in DEPTHWISE_CONV2D

The for_each already runs over 0 <= m < M, we don't
need c to also iterate over M.

Change-Id: Ia72b4be7e68e2f210e9bfc75e3958089d9930c47
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
---
 chapters/tensor_ops.adoc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/chapters/tensor_ops.adoc b/chapters/tensor_ops.adoc
index b2f0754..9a4ab88 100644
--- a/chapters/tensor_ops.adoc
+++ b/chapters/tensor_ops.adoc
@@ -294,7 +294,7 @@ ERROR_IF(pad_top < 0 || pad_bottom < 0 || pad_left < 0 || pad_right < 0);
 ERROR_IF(stride_y < 1 || stride_x < 1);
 ERROR_IF(dilation_y < 1 || dilation_x < 1);
 pad = flatten([0,0], pad, [0,0]);
-for_each(0 <= n<N, 0 <= oy < H, 0 <= ox < W; 0 <= c < (C * M), 0 <= m < M) {
+for_each(0 <= n<N, 0 <= oy < H, 0 <= ox < W; 0 <= c < C, 0 <= m < M) {
     out_t acc = 0;
     iy = oy * stride_y - pad_top;
     ix = ox * stride_x - pad_left;
-- 
cgit v1.2.1