diff options
| -rw-r--r-- | chapters/tensor_ops.adoc | 2 | ||||
| -rw-r--r-- | tosa.xml | 4 |
2 files changed, 1 insertions, 5 deletions
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<out_t>(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, @@ -620,10 +620,6 @@ <levellimit value="stride_x" limit="MAX_STRIDE"/> <rank min="1" max="1"/> </argument> - <argument category="attribute" name="out_shape" type="tensor_t" shape="[4]" tensor-element-type="i32_t"> - <description>[N,OH,OW,OC]</description> - <rank min="1" max="1"/> - </argument> <argument category="attribute" name="input_zp" type="tensor_t" shape="-" tensor-element-type="in_t"> <description>Input tensor zero point. Must be zero for non-int8 types.</description> <rank min="0" max="0"/> |