aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorEric Kunze <eric.kunze@arm.com>2022-01-07 14:27:01 -0800
committerEric Kunze <eric.kunze@arm.com>2022-01-07 14:28:41 -0800
commite942f07388abbdb3b66974a4f41cf9d21e56a4dc (patch)
tree31852a3cd1f88f00e95ad3b7aa483bd9ee721213
parent2a3d560999c794e8faacefa93db23bf1e1714532 (diff)
downloadspecification-e942f07388abbdb3b66974a4f41cf9d21e56a4dc.tar.gz
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>
-rw-r--r--chapters/tensor_ops.adoc2
1 files changed, 1 insertions, 1 deletions
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;