diff options
author | Eric Kunze <eric.kunze@arm.com> | 2024-03-19 16:25:09 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-04-12 15:26:20 -0700 |
commit | 8301291be25af9a66c16b29a9231a7b814867ddc (patch) | |
tree | b3b6d212026b70d54970b6367b0ea4f3e82fae00 | |
parent | ce015e7d72f522ac310404f29a0bc467f9f69c58 (diff) | |
download | specification-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.adoc | 11 | ||||
-rw-r--r-- | chapters/shape.adoc | 11 | ||||
-rw-r--r-- | 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. @@ -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="-"> |