Define TENSORFLOW_SUM in NeuralNetworksEx.h (#3578)
authorPrasanna R/System SW /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Wed, 14 Nov 2018 07:35:04 +0000 (13:05 +0530)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Wed, 14 Nov 2018 07:35:04 +0000 (16:35 +0900)
This patch defines TENSORFLOW_SUM in NeuralNetworksEx.h.
Related issue: #3347.

Signed-off-by: prasannar <prasanna.r@samsung.com>
include/NeuralNetworksEx.h

index e68fa54..0b5366c 100644 (file)
@@ -172,6 +172,25 @@ typedef enum {
      * * 0: The output tensor, of the same {@link OperandCode} and shape as input0.
      */
     ANEURALNETWORKS_EXP_EX = 50008,
+
+    /**
+     * Computes the sum of elements across dimensions of a tensor.
+     *
+     * Reduces the input tensor along the given dimensions to reduce.
+     *
+     * Supported tensor {@link OperandCode}:
+     * * {@link ANEURALNETWORKS_TENSOR_FLOAT32}
+     *
+     * Supported tensor rank: up to 4
+     *
+     * Inputs:
+     * * 0: A tensor, specifying the input.
+     * * 1: A 1-D Tensor of {@link ANEURALNETWORKS_TENSOR_INT32}. The dimensions
+     *      to reduce.
+     *
+     * Outputs:
+     * * 0: A tensor of the same {@link OperandCode} as input0.
+     */
     ANEURALNETWORKS_TENSORFLOW_SUM_EX = 50009,
     ANEURALNETWORKS_TRANSPOSE_CONV_EX = 50010,