diff options
author | Eric Kunze <eric.kunze@arm.com> | 2021-03-12 10:27:39 -0800 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2021-03-16 16:41:11 -0700 |
commit | 67ed038545b0495124b5d5a30786cc4180dd6a9c (patch) | |
tree | 93a59ad31e2e6125f4a060259e8bf3af8b7cc464 /chapters/reduction.adoc | |
parent | 839830a72ebb27c4a818e15d3334e79085906dc8 (diff) | |
download | specification-67ed038545b0495124b5d5a30786cc4180dd6a9c.tar.gz |
Note limits on axis parameter
This also adds a restriction on CONCAT to have all inputs be the
same rank.
Update the ARGMAX pseudocode to handle corner cases properly and
use axis/rank consistent with other operators.
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: I941ac172ee79424ac04d562cedebb75da76659a5
Diffstat (limited to 'chapters/reduction.adoc')
-rw-r--r-- | chapters/reduction.adoc | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/chapters/reduction.adoc b/chapters/reduction.adoc index e605386..b84b7f0 100644 --- a/chapters/reduction.adoc +++ b/chapters/reduction.adoc @@ -19,7 +19,7 @@ Reduce a tensor along the given axis with a logical AND operation |Argument|Type|Name|Shape|Description |Input|in_t*|input|shape1|Input tensor with rank from 1 to 4 -|Attribute|int32_t|axis|-|Axis to reduce +|Attribute|int32_t|axis|-|Axis to reduce, in range from 0 to rank(shape1)-1 |Output|in_t*|output|shape|Output tensor. Same rank as the input tensor. |=== @@ -60,7 +60,7 @@ Reduce a tensor along the given axis with a logical OR operation |Argument|Type|Name|Shape|Description |Input|in_t*|input|shape1|Input tensor with rank from 1 to 4 -|Attribute|int32_t|axis|-|Axis to reduce +|Attribute|int32_t|axis|-|Axis to reduce, in range from 0 to rank(shape1)-1 |Output|in_t*|output|shape|Output tensor. Same rank as the input tensor. |=== @@ -101,7 +101,7 @@ Reduce a tensor along the given axis with a maximum operation |Argument|Type|Name|Shape|Description |Input|in_t*|input|shape1|Input tensor with rank from 1 to 4 -|Attribute|int32_t|axis|-|Axis to reduce +|Attribute|int32_t|axis|-|Axis to reduce, in range from 0 to rank(shape1)-1 |Output|in_t*|output|shape|Output tensor. Same rank as the input tensor. |=== @@ -144,7 +144,7 @@ Reduce a tensor along the given axis with a minimum operation |Argument|Type|Name|Shape|Description |Input|in_t*|input|shape1|Input tensor with rank from 1 to 4 -|Attribute|int32_t|axis|-|Axis to reduce +|Attribute|int32_t|axis|-|Axis to reduce, in range from 0 to rank(shape1)-1 |Output|in_t*|output|shape|Output tensor. Same rank as the input tensor. |=== @@ -192,7 +192,7 @@ Reduce a tensor along the given axis by computing the product of the axis. |Argument|Type|Name|Shape|Description |Input|in_t*|input|shape1|Input tensor with rank from 1 to 4 -|Attribute|int32_t|axis|-|Axis to reduce +|Attribute|int32_t|axis|-|Axis to reduce, in range from 0 to rank(shape1)-1 |Output|in_t*|output|shape|Output tensor. Same rank as the input tensor. |=== @@ -233,7 +233,7 @@ Reduce a tensor along the given axis by computing the sum of the axis. |Argument|Type|Name|Shape|Description |Input|in_t*|input|shape1|Input tensor with rank from 1 to 4 -|Attribute|int32_t|axis|-|Axis to reduce +|Attribute|int32_t|axis|-|Axis to reduce, in range from 0 to rank(shape1)-1 |Output|in_t*|output|shape|Output tensor. Same rank as the input tensor. |=== |