From 8301291be25af9a66c16b29a9231a7b814867ddc Mon Sep 17 00:00:00 2001
From: Eric Kunze <eric.kunze@arm.com>
Date: Tue, 19 Mar 2024 16:25:09 -0700
Subject: 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
---
 chapters/data_layout.adoc | 11 ------
 chapters/shape.adoc       | 11 ++++++
 tosa.xml                  | 96 +++++++++++++++++++++++------------------------
 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
@@ -2202,54 +2202,6 @@
           <op_profile name="MI"/>
         </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>
@@ -3236,6 +3188,54 @@ used.</description>
           <op_profile name="MT"/>
         </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>
-- 
cgit v1.2.1