From def882e8363e887ce53e4b5c34bdc3ba3dfbc55c Mon Sep 17 00:00:00 2001 From: Dominic Symes Date: Thu, 21 Sep 2023 10:49:28 +0100 Subject: TRANSPOSE_CONV2D: Remove redundant out_shape argument The output shape can be calculated from the input shape and attributes so the out_shape argument is redundant. Change-Id: I7e6b0b7f03dc75ccdf967261fad4185c513ed187 Signed-off-by: Dominic Symes --- chapters/tensor_ops.adoc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'chapters') diff --git a/chapters/tensor_ops.adoc b/chapters/tensor_ops.adoc index b9d54c1..b3de433 100644 --- a/chapters/tensor_ops.adoc +++ b/chapters/tensor_ops.adoc @@ -402,7 +402,7 @@ ERROR_IF(OH != (IH - 1) * stride_y + out_pad_top + out_pad_bottom + KH); ERROR_IF(OW != (IW - 1) * stride_x + out_pad_left + out_pad_right + KW); ERROR_IF(BC != OC && BC != 1); -for_each(index in out_shape) { +for_each(index in [N, OH, OW, OC]) { tensor_write(output, [N,OH,OW,OC], index, bias[(BC == 1) ? 0 : index[3]]) } for_each(0 <= n < N, 0 <= iy < IH, 0 <= ix < IW, 0 <= oc < OC, -- cgit v1.2.1