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/data_layout.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/data_layout.adoc')
-rw-r--r-- | chapters/data_layout.adoc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/chapters/data_layout.adoc b/chapters/data_layout.adoc index 4b01d55..de90322 100644 --- a/chapters/data_layout.adoc +++ b/chapters/data_layout.adoc @@ -18,8 +18,8 @@ No data conversion happens during a concat operation. |=== |Argument|Type|Name|Shape|Description -|Input|in_t**|input1|shapes1[]|List of input tensors -|Attribute|int|axis|-|Axis along which concatenation is to occur +|Input|in_t**|input1|shapes1[]|List of input tensors. All inputs must have the same rank +|Attribute|int|axis|-|Axis along which concatenation is to occur, in range from 0 to rank(shape)-1 |Output|in_t*|output|shape|Output tensor |=== @@ -149,7 +149,7 @@ Returns a tensor with the same type/values as the input, with the data reversed |Argument|Type|Name|Shape|Description |Input|in_t*|input|shape|Input tensor from 1 to 4 dims -|Attribute|int|axis|-|Axis to reverse +|Attribute|int|axis|-|Axis to reverse, in range from 0 to rank(shape)-1 |Output|in_t*|output|shape|Output tensor. Same shape as input tensor. |=== |