diff options
author | Dominic Symes <dominic.symes@arm.com> | 2022-07-27 12:53:11 +0100 |
---|---|---|
committer | Dominic Symes <dominic.symes@arm.com> | 2022-08-26 09:11:01 +0100 |
commit | 8783adb282f4d6414a1442042bd8e02afcd12b18 (patch) | |
tree | 82aad31c55fce13032f675e4acb2c4329861184d /tosa.xml | |
parent | eda7b126d3914e9461cf014439b3571b9e6a9c41 (diff) | |
download | specification-8783adb282f4d6414a1442042bd8e02afcd12b18.tar.gz |
RESCALE and POW: remove rank restriction
Also be consistent to use the term 'rank'
rather than 'dims'.
Change-Id: Ica827fcfd44e9735da2d1a4ddb8cb1fa04c6479b
Signed-off-by: Dominic Symes <dominic.symes@arm.com>
Diffstat (limited to 'tosa.xml')
-rw-r--r-- | tosa.xml | 8 |
1 files changed, 4 insertions, 4 deletions
@@ -901,7 +901,7 @@ <name>POW</name> <arguments> <argument category="input" name="input1" type="in_out_t*" shape="shape1"> - <description>Input tensor from 1 to 4 dims</description> + <description>Input tensor</description> </argument> <argument category="input" name="input2" type="in_out_t*" shape="shape2"> <description>Input tensor with the same rank as input1</description> @@ -1650,7 +1650,7 @@ <name>REVERSE</name> <arguments> <argument category="input" name="input" type="in_out_t*" shape="shape"> - <description>Input tensor from 1 to 4 dims</description> + <description>Input tensor with rank from 1 to 4</description> </argument> <argument category="attribute" name="axis" type="int32_t" shape="-"> <description>Axis to reverse, in range from 0 to rank(shape)-1</description> @@ -2004,7 +2004,7 @@ used.</description> <name>RESCALE</name> <arguments> <argument category="input" name="input" type="in_t" shape="shape"> - <description>Input tensor from 1 to 4 dims</description> + <description>Input tensor</description> </argument> <argument category="output" name="output" type="out_t" shape="shape"> <description>Output tensor with the same shape as input</description> @@ -2028,7 +2028,7 @@ used.</description> <description>Select double round mode</description> </argument> <argument category="attribute" name="per_channel" type="bool_t" shape="-"> - <description>if (per_channel) NC=shape[dims-1] else NC=1</description> + <description>if (per_channel) NC=shape[rank(shape)-1] else NC=1</description> </argument> </arguments> <types> |