Defining EQUAL_EX in NeuralNetworksEx.h (#3354)

This patch defines EQUAL_EX in NeuralNetworksEx.h
Related issue: #3347

Signed-off-by: prasannar <prasanna.r@samsung.com>

diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h
index 1778621..7213198 100644 (file)
--- a/include/NeuralNetworksEx.h
+++ b/include/NeuralNetworksEx.h
@@ -43,6 +43,39 @@ typedef enum {
     ANEURALNETWORKS_EXP_EX = 50008,
     ANEURALNETWORKS_TENSORFLOW_SUM_EX = 50009,
     ANEURALNETWORKS_TRANSPOSE_CONV_EX = 50010,
+
+    /**
+     * Returns the truth value of (x == y) element-wise.
+     *
+     * Takes two input tensors of identical {@link OperandCode} and compatible dimensions.
+     * The output is the result of comparison of two input tensors.
+     *
+     * Two dimensions are compatible when:
+     *     1. they are equal, or
+     *     2. one of them is 1
+     *
+     * The size of the output is the maximum size along each dimension of the
+     * input operands. It starts with the trailing dimensions, and works its way
+     * forward.
+     *
+     * Supported tensor {@link OperandCode}:
+     * * {@link ANEURALNETWORKS_TENSOR_FLOAT32}
+     * * {@link ANEURALNETWORKS_TENSOR_INT32}
+     * * {@link ANEURALNETWORKS_TENSOR_QUANT8_ASYMM}
+     *
+     * Supported tensor rank: up to 4
+     *
+     * Inputs:
+     * * 0: An n-D tensor, specifying the first input.
+     * * 1: A tensor of the same {@link OperandCode}, and compatible dimensions
+     *      as input0.
+     *
+     * Outputs:
+     * * 0: A boolean tensor indicating the truth value of (x == y)
+     *      Stored as {@link ANEURALNETWORKS_TENSOR_QUANT8_ASYMM} with offset 0
+     *      and scale 1.0f.
+     *      A non-zero byte represents True, a hit. A zero indicates otherwise.
+     */
     ANEURALNETWORKS_EQUAL_EX = 50011,
     ANEURALNETWORKS_ABS_EX = 50012,
     ANEURALNETWORKS_PACK_EX = 50013,