Add comment for argmax_ex (#3576)

Add comment for ANEURALNETWORKS_ARGMAX_EX in NeuralNetworksEx.h

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>

diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h
index 0b5366c..89d25db 100644 (file)
--- 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,
 
     /**