diff options
author | Eric Kunze <eric.kunze@arm.com> | 2020-10-01 18:50:46 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2020-10-01 18:50:46 -0700 |
commit | 3309a5362a13f840e84a2f67b9ba7141aae58cc4 (patch) | |
tree | 74bde543bc11bf43208fa4b3789e2def23919890 /chapters/scatter_gather.adoc | |
parent | c92cb063a2ebfad2fd9ffd4c4a73184327b864ed (diff) | |
download | specification-3309a5362a13f840e84a2f67b9ba7141aae58cc4.tar.gz |
TOSA spec 0.20.0, initial public versionv0.20.0
Change-Id: I7c528ebf7e78759ada4bb951b2471c367d3f4525
Diffstat (limited to 'chapters/scatter_gather.adoc')
-rw-r--r-- | chapters/scatter_gather.adoc | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/chapters/scatter_gather.adoc b/chapters/scatter_gather.adoc new file mode 100644 index 0000000..29ba391 --- /dev/null +++ b/chapters/scatter_gather.adoc @@ -0,0 +1,40 @@ +// +// This confidential and proprietary software may be used only as +// authorised by a licensing agreement from ARM Limited +// (C) COPYRIGHT 2020 ARM Limited +// ALL RIGHTS RESERVED +// The entire notice above must be reproduced on all authorised +// copies and copies may only be made to the extent permitted +// by a licensing agreement from ARM Limited. + +=== Scatter/Gather Operators + +==== GATHER + +Generate a tensor for which each element in the output is a subtensor of the values tensor along the given axis, based on the value of indices. + +*Arguments:* + +|=== +|Argument|Type|Name|Shape|Description + +|Input|index_t*|indices|shape|Input tensor from 1 to 4 dims +|Input|value_t*|values|shape|Input tensor from 1 to 4 dims. Must be of rank at least axis + 1 +|Attribute|int|axis|-|Scalar value denoting which dimension to be used for striding. +|Output|out_t*|output|shape|Output tensor +|=== + +*Quantization Parameters:* + +None + +*Supported Data Types:* + +|=== +|Profile|Mode|index_t|value_t|out_t + +|Any|signed 8|int8/int16|aint8|aint8 +|Any|signed 16|int8/int16|int16|int16 +|Any|signed 32|int8/int16|int32|int32 +|=== + |