aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorEric Kunze <eric.kunze@arm.com>2024-03-19 16:25:09 -0700
committerEric Kunze <eric.kunze@arm.com>2024-04-12 15:26:20 -0700
commit8301291be25af9a66c16b29a9231a7b814867ddc (patch)
treeb3b6d212026b70d54970b6367b0ea4f3e82fae00
parentce015e7d72f522ac310404f29a0bc467f9f69c58 (diff)
downloadspecification-8301291be25af9a66c16b29a9231a7b814867ddc.tar.gz
Move DIM to Shape operator section
Is more appropriate than the Data Layout section Signed-off-by: Eric Kunze <eric.kunze@arm.com> Change-Id: I5427d867b13a69dc0d66e268db4c64e047c65c7f
-rw-r--r--chapters/data_layout.adoc11
-rw-r--r--chapters/shape.adoc11
-rw-r--r--tosa.xml96
3 files changed, 59 insertions, 59 deletions
diff --git a/chapters/data_layout.adoc b/chapters/data_layout.adoc
index c6afc4f..7b98271 100644
--- a/chapters/data_layout.adoc
+++ b/chapters/data_layout.adoc
@@ -33,17 +33,6 @@ include::{generated}/operators/PAD.adoc[]
include::{pseudocode}/operators/PAD.tosac[lines=10..-1]
----
-==== DIM
-
-Returns a length 1 shape_t of the size of the input tensor for the given axis.
-
-include::{generated}/operators/DIM.adoc[]
-
-[source,c++]
-----
-include::{pseudocode}/operators/DIM.tosac[lines=10..-1]
-----
-
==== RESHAPE
Returns a tensor with the same type/values as the input, with a new shape specified by the shape argument. Reshape may operate on tensors of any rank. No data conversion happens during a reshape operation.
diff --git a/chapters/shape.adoc b/chapters/shape.adoc
index 1b58465..1f91cd4 100644
--- a/chapters/shape.adoc
+++ b/chapters/shape.adoc
@@ -49,6 +49,17 @@ include::{generated}/operators/CONST_SHAPE.adoc[]
include::{pseudocode}/operators/CONST_SHAPE.tosac[lines=10..-1]
----
+==== DIM
+
+Returns a length 1 shape_t of the size of the input tensor for the given axis.
+
+include::{generated}/operators/DIM.adoc[]
+
+[source,c++]
+----
+include::{pseudocode}/operators/DIM.tosac[lines=10..-1]
+----
+
==== DIV_SHAPE
Elementwise integer divide of input1 by input2. The result of the divide is truncated towards zero.
diff --git a/tosa.xml b/tosa.xml
index 5973ff1..0b4315b 100644
--- a/tosa.xml
+++ b/tosa.xml
@@ -2203,54 +2203,6 @@
</typesupport>
</operator>
<operator>
- <name>DIM</name>
- <arguments>
- <argument category="input" name="input1" type="tensor_t" shape="shape" tensor-element-type="in_t">
- <description>Input tensor</description>
- <levellimit value="rank(shape)" limit="MAX_RANK"/>
- <rank min="1" max="MAX_RANK"/>
- </argument>
- <argument category="attribute" name="axis" type="tensor_t" shape="-" tensor-element-type="i32_t">
- <description>Axis in range from 0 to rank(shape) - 1</description>
- <rank min="0" max="0"/>
- </argument>
- <argument category="output" name="output" type="shape_t" shape="-" tensor-element-type="-" >
- <description>Output shape type of size 1 giving the size of the shape for the given axis</description>
- </argument>
- </arguments>
- <types>
- <type name='in_t'/>
- </types>
- <typesupport mode="boolean" in_t="bool_t">
- <op_profile name="BI"/>
- <op_profile name="MI"/>
- </typesupport>
- <typesupport mode="signed 8" in_t="i8_t">
- <op_profile name="BI"/>
- </typesupport>
- <typesupport mode="signed 16" in_t="i16_t">
- <op_profile name="BI"/>
- </typesupport>
- <typesupport mode="signed 32" in_t="i32_t">
- <op_profile name="BI"/>
- </typesupport>
- <typesupport mode="fp8e4m3" in_t="fp8e4m3_t">
- <op_profile name="EXT-FP8E4M3"/>
- </typesupport>
- <typesupport mode="fp8e5m2" in_t="fp8e5m2_t">
- <op_profile name="EXT-FP8E5M2"/>
- </typesupport>
- <typesupport mode="fp16" in_t="fp16_t">
- <op_profile name="MI"/>
- </typesupport>
- <typesupport mode="bf16" in_t="bf16_t">
- <op_profile name="EXT-BF16"/>
- </typesupport>
- <typesupport mode="fp32" in_t="fp32_t">
- <op_profile name="MI"/>
- </typesupport>
- </operator>
- <operator>
<name>RESHAPE</name>
<arguments>
<argument category="input" name="input1" type="tensor_t" shape="shape1" tensor-element-type="in_out_t">
@@ -3237,6 +3189,54 @@ used.</description>
</typesupport>
</operator>
<operator>
+ <name>DIM</name>
+ <arguments>
+ <argument category="input" name="input1" type="tensor_t" shape="shape" tensor-element-type="in_t">
+ <description>Input tensor</description>
+ <levellimit value="rank(shape)" limit="MAX_RANK"/>
+ <rank min="1" max="MAX_RANK"/>
+ </argument>
+ <argument category="attribute" name="axis" type="tensor_t" shape="-" tensor-element-type="i32_t">
+ <description>Axis in range from 0 to rank(shape) - 1</description>
+ <rank min="0" max="0"/>
+ </argument>
+ <argument category="output" name="output" type="shape_t" shape="-" tensor-element-type="-" >
+ <description>Output shape type of size 1 giving the size of the shape for the given axis</description>
+ </argument>
+ </arguments>
+ <types>
+ <type name='in_t'/>
+ </types>
+ <typesupport mode="boolean" in_t="bool_t">
+ <op_profile name="BI"/>
+ <op_profile name="MI"/>
+ </typesupport>
+ <typesupport mode="signed 8" in_t="i8_t">
+ <op_profile name="BI"/>
+ </typesupport>
+ <typesupport mode="signed 16" in_t="i16_t">
+ <op_profile name="BI"/>
+ </typesupport>
+ <typesupport mode="signed 32" in_t="i32_t">
+ <op_profile name="BI"/>
+ </typesupport>
+ <typesupport mode="fp8e4m3" in_t="fp8e4m3_t">
+ <op_profile name="EXT-FP8E4M3"/>
+ </typesupport>
+ <typesupport mode="fp8e5m2" in_t="fp8e5m2_t">
+ <op_profile name="EXT-FP8E5M2"/>
+ </typesupport>
+ <typesupport mode="fp16" in_t="fp16_t">
+ <op_profile name="MI"/>
+ </typesupport>
+ <typesupport mode="bf16" in_t="bf16_t">
+ <op_profile name="EXT-BF16"/>
+ </typesupport>
+ <typesupport mode="fp32" in_t="fp32_t">
+ <op_profile name="MI"/>
+ </typesupport>
+ </operator>
+ <operator>
<name>DIV_SHAPE</name>
<arguments>
<argument category="input" name="input1" type="shape_t" shape="-" tensor-element-type="-">