diff options
author | Vishal Keshav/SNAP /SRI-Bangalore/Engineer/삼성전자 <vishal.k1@samsung.com> | 2018-11-23 08:08:40 +0530 |
---|---|---|
committer | 오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com> | 2018-11-23 11:38:40 +0900 |
commit | 0a6466beb542dbf749ffb719315c32e67f72bec1 (patch) | |
tree | 0e1a6ad4eec1ac5b1a7948964400c8065ccc4c10 /include | |
parent | bc154fc03704de9bf5b8c05cb243f1b11c664ebf (diff) | |
download | nnfw-0a6466beb542dbf749ffb719315c32e67f72bec1.tar.gz nnfw-0a6466beb542dbf749ffb719315c32e67f72bec1.tar.bz2 nnfw-0a6466beb542dbf749ffb719315c32e67f72bec1.zip |
Defined CUSTOM pack operation (#3648)
Added details for pack operation
Signed-off-by: Vishal Keshav <vishal.k1@samsung.com>
Diffstat (limited to 'include')
-rw-r--r-- | include/NeuralNetworksEx.h | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h index 45e6167c2..9ff82917a 100644 --- a/include/NeuralNetworksEx.h +++ b/include/NeuralNetworksEx.h @@ -228,6 +228,31 @@ typedef enum { */ ANEURALNETWORKS_EQUAL_EX = 50011, ANEURALNETWORKS_ABS_EX = 50012, + /** + * Packs a list of rank-R tensors into one rank- (R+1) tensor along the axis dimension. + * + * The input tensors must have identical {@link OperandCode} and the same + * dimensions. + * + * Supported tensor {@link OperandCode}: + * * {@link ANEURALNETWORKS_TENSOR_FLOAT32} + * * {@link ANEURALNETWORKS_TENSOR_QUANT8_ASYMM} + * + * Supported tensor rank: up to 3 + * + * Inputs: + * * 0 ~ n-1: The list of n input tensors, of shape + * [D0, D1, ..., Daxis(i), ..., Dm]. For inputs of + * {@link ANEURALNETWORKS_TENSOR_QUANT8_ASYMM}, all input tensors + * must have the same scale and zeroPoint. + * * n: An {@link ANEURALNETWORKS_INT32} scalar, specifying the + * pack axis. + * + * Outputs: + * * 0: The output, a tensor of the same {@link OperandCode} as the input + * tensors. The output shape is [D0, D1, ..., N at Daxis(i), ..., Dm+1] + * where N is the number of tensors to be packed. + */ ANEURALNETWORKS_PACK_EX = 50013, /** * Unpacks a given rank-R tensors into num_splits rank- (R-1) tensors along the axis dimention. |