From: Vishal Keshav/SNAP /SRI-Bangalore/Engineer/삼성전자 Date: Fri, 23 Nov 2018 02:38:40 +0000 (+0530) Subject: Defined CUSTOM pack operation (#3648) X-Git-Tag: 0.3~344 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=0a6466beb542dbf749ffb719315c32e67f72bec1;p=platform%2Fcore%2Fml%2Fnnfw.git Defined CUSTOM pack operation (#3648) Added details for pack operation Signed-off-by: Vishal Keshav --- diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h index 45e6167..9ff8291 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.