diff options
author | Eric Kunze <eric.kunze@arm.com> | 2022-05-19 14:50:18 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2022-06-13 20:23:28 +0000 |
commit | a177e435e4065f68b5c8c2cc3e84a2e4f7d1ecae (patch) | |
tree | 18a3f03e63a0ececaae76e8300cab3d15c163ac3 /chapters/tensor_ops.adoc | |
parent | 0738025d386e0aa0e274ded20afdf000b22a1635 (diff) | |
download | specification-a177e435e4065f68b5c8c2cc3e84a2e4f7d1ecae.tar.gz |
Fix typo on CONV3D pseudocode
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: Ib953cc4b635469034300662b9ae997d0a1bd58a7
Diffstat (limited to 'chapters/tensor_ops.adoc')
-rw-r--r-- | chapters/tensor_ops.adoc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/chapters/tensor_ops.adoc b/chapters/tensor_ops.adoc index bd5ffde..8bcb115 100644 --- a/chapters/tensor_ops.adoc +++ b/chapters/tensor_ops.adoc @@ -244,7 +244,7 @@ for_each(0 <= n < N, 0 <= od < OD, 0 <= oy < OH, 0 <= ox < OW; 0 <= oc < OC) { d = id + kd * dilation_d; y = iy + ky * dilation_y; x = ix + kx * dilation_x; - if (0 <= x < IW && 0 <= y < IH && 0 <= d <= ID) { + if (0 <= x < IW && 0 <= y < IH && 0 <= d < ID) { out_t value = tensor_read<in_t>(input, [N,ID,IH,IW,IC], [n,d,y,x,ic]); out_t weight = tensor_read<weight_t>(weight,[OC,KD,KH,KW,IC],[oc,kd,ky,kx,ic]); value = value - input_zp; @@ -302,7 +302,7 @@ ERROR_IF(OH != idiv_check(IH - 1 + pad_top + pad_bottom - (KH - 1) * dilation_y, ERROR_IF(OW != idiv_check(IW - 1 + pad_left + pad_right - (KW - 1) * dilation_x, stride_x) + 1); pad = flatten([0,0], pad, [0,0]); -for_each(0 <= n<N, 0 <= oy < OH, 0 <= ox < OW; 0 <= c < C, 0 <= m < M) { +for_each(0 <= n < N, 0 <= oy < OH, 0 <= ox < OW; 0 <= c < C, 0 <= m < M) { out_t acc = 0; iy = oy * stride_y - pad_top; ix = ox * stride_x - pad_left; |