diff options
author | Eric Kunze <eric.kunze@arm.com> | 2021-10-25 16:13:22 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2021-10-28 09:57:55 -0700 |
commit | 82f19e2ad25bcbdde8e7f8b6bd6a6064a207fe36 (patch) | |
tree | c7a685d03bf2a9dd5a171b6c7409baf8c687da4f /chapters/ewise_ternary.adoc | |
parent | 3fe5256340778f586b80ac02b0632b54a39723f1 (diff) | |
download | specification-82f19e2ad25bcbdde8e7f8b6bd6a6064a207fe36.tar.gz |
Readability fixes for pseudocode
Avoid use of acc for variables when they are not convolution
accumulators.
Use argument types appropriately.
Add missing pseudocode for some MI operators
Change-Id: I9113f9228dbcafb85206bcc39310e9599cb12c08
Diffstat (limited to 'chapters/ewise_ternary.adoc')
-rw-r--r-- | chapters/ewise_ternary.adoc | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/chapters/ewise_ternary.adoc b/chapters/ewise_ternary.adoc index ecf40d1..c6babbc 100644 --- a/chapters/ewise_ternary.adoc +++ b/chapters/ewise_ternary.adoc @@ -18,7 +18,7 @@ Elementwise select of the output based on a condition. |=== |Argument|Type|Name|Shape|Description -|Input|bool_t|input1|shape1|Input selector tensor +|Input|cmp_t|input1|shape1|Input selector tensor |Input|in_t*|input2|shape2|Input value tensor if input1 is True |Input|in_t*|input3|shape3|Input value tensor if input1 is False |Output|in_t*|output|shape|Output tensor of same type as input2 and input3, with broadcast shape if necessary @@ -32,22 +32,22 @@ for_each(index in shape) { index1 = apply_broadcast(shape, shape1, index); index2 = apply_broadcast(shape, shape2, index); index3 = apply_broadcast(shape, shpae3, index); - bool_t value1 = tensor_read<in_t>(input1, shape1, index1); + cmp_t value1 = tensor_read<cmp_t>(input1, shape1, index1); in_t value2 = tensor_read<in_t>(input2, shape2, index2); in_t value3 = tensor_read<in_t>(input3, shape3, index3); - in_t acc = 0; - if (value1 == True) { - acc = value2; + in_t result; + if (value1) { + result = value2; } else { - acc = value3; + result = value3; } - tensor_write<in_t>(output, shape, index, acc); + tensor_write<in_t>(output, shape, index, result); } ---- *Supported Data Types:* |=== -|Profile|Mode|bool_t|in_t +|Profile|Mode|cmp_t|in_t |Any|Boolean|bool_t|bool_t |Any|signed 8|bool_t|int8_t |