From f01dc2338115706374a16d8f620fecafb782b6cb Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=98=A4=ED=98=95=EC=84=9D/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Wed, 14 Nov 2018 18:02:43 +0900 Subject: [PATCH] Add comment for argmax_ex (#3576) Add comment for ANEURALNETWORKS_ARGMAX_EX in NeuralNetworksEx.h Signed-off-by: Hyeongseok Oh --- include/NeuralNetworksEx.h | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h index 0b5366c..89d25db 100644 --- a/include/NeuralNetworksEx.h +++ b/include/NeuralNetworksEx.h @@ -230,6 +230,25 @@ typedef enum { ANEURALNETWORKS_ABS_EX = 50012, ANEURALNETWORKS_PACK_EX = 50013, ANEURALNETWORKS_UNPACK_EX = 50014, + + /** + * Find index with the largest value across axes of a input tensor. + * + * Reduces the input tensor along the given dimensions to reduce. + * + * Supported tensor {@link OperandCode}: + * * {@link ANEURALNETWORKS_TENSOR_FLOAT32} + * * {@link ANEURALNETWORKS_TENSOR_QUANT8_ASYMM} + * + * Inputs: + * * 0: An n-D tensor, specifying the input. + * * 1: A 1-D Tensor of {@link ANEURALNETWORKS_TENSOR_INT32}. The dimensions + * to reduce. Must be in the range [-rank(input_tensor), rank(input_tensor)). + * + * Outputs: + * * 0: A output tensor of {@link ANEURALNETWORKS_TENSOR_INT32}. + * The rank of output tensor should be rank(input_tensor)-1. + */ ANEURALNETWORKS_ARGMAX_EX = 50015, /** -- 2.7.4