From 8783adb282f4d6414a1442042bd8e02afcd12b18 Mon Sep 17 00:00:00 2001
From: Dominic Symes <dominic.symes@arm.com>
Date: Wed, 27 Jul 2022 12:53:11 +0100
Subject: 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>
---
 tosa.xml | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'tosa.xml')

diff --git a/tosa.xml b/tosa.xml
index 6674ef0..bc86354 100644
--- a/tosa.xml
+++ b/tosa.xml
@@ -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>
-- 
cgit v1.2.1