From 3309a5362a13f840e84a2f67b9ba7141aae58cc4 Mon Sep 17 00:00:00 2001 From: Eric Kunze Date: Thu, 1 Oct 2020 18:50:46 -0700 Subject: TOSA spec 0.20.0, initial public version Change-Id: I7c528ebf7e78759ada4bb951b2471c367d3f4525 --- chapters/scatter_gather.adoc | 40 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 chapters/scatter_gather.adoc (limited to 'chapters/scatter_gather.adoc') 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 +|=== + -- cgit v1.2.1