diff options
| author | Eric Kunze <eric.kunze@arm.com> | 2020-11-04 14:23:31 -0800 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2020-11-06 09:10:10 -0800 |
| commit | 646ef42f0357c85b2898c39d1657a85487ac751c (patch) | |
| tree | e94d322939b977c9a1037b1a8a47e2a844a85687 /chapters/ewise_ternary.adoc | |
| parent | 9424cc4d4c5002c1da441f20052342b6b1b6ac7c (diff) | |
| download | specification-646ef42f0357c85b2898c39d1657a85487ac751c.tar.gz | |
Operator argument consistency improvement.
Add values attribute to CONST operator.
Remove stale references to input tensor 0 and 1.
Remove out_t types where in and out are guaranteed
to be the same type.
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: I75a8eb4dee67afbee3652d9e937aa0b82318dbd0
Diffstat (limited to 'chapters/ewise_ternary.adoc')
| -rw-r--r-- | chapters/ewise_ternary.adoc | 27 |
1 files changed, 14 insertions, 13 deletions
diff --git a/chapters/ewise_ternary.adoc b/chapters/ewise_ternary.adoc index 1724777..8dc9172 100644 --- a/chapters/ewise_ternary.adoc +++ b/chapters/ewise_ternary.adoc @@ -18,10 +18,10 @@ Elementwise select of the output based on a condition. |=== |Argument|Type|Name|Shape|Description -|Input|bool_t|input1|shape1| -|Input|in_t*|input2|shape2|Input tensor from 1 to 4 dims -|Input|in_t*|input3|shape3|Input tensor with the same rank as Input 0 -|Output|out_t*|output|shape|Output tensor of same type, as the input tensors, with broadcast shape if necessary +|Input|bool_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 |=== *Quantization Parameters:* @@ -39,22 +39,23 @@ for_each (index in shape) { bool_t value1 = tensor_read<in_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){ - in_t acc = value2 + acc = value2 } else { - in_t acc = value3 + acc = value3 } - tensor_write<out_t>(output, shape, index, acc) + tensor_write<in_t>(output, shape, index, acc) } ---- *Supported Data Types:* |=== -|Profile|Mode|bool_t|in_t|out_t +|Profile|Mode|bool_t|in_t -|Any|Boolean|bool|bool|bool -|Any|signed 8|bool|aint8/int8|aint8/int8 -|Any|signed 16|bool|int16|int16 -|Any|signed 32|bool|int32|int32 -|MI, MT|float|bool|float|float +|Any|Boolean|bool|bool +|Any|signed 8|bool|aint8/int8 +|Any|signed 16|bool|int16 +|Any|signed 32|bool|int32 +|MI, MT|float|bool|float |=== |