Defining RSQRT_EX in NeuralNetworksEx.h (#3415)

This patch defines RSQRT_EX in NeuralNetworksEx.h

Signed-off-by: shubham <shub98.gupta@samsung.com>

diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h
index a61ca64..3fe9ec8 100644 (file)
--- a/include/NeuralNetworksEx.h
+++ b/include/NeuralNetworksEx.h
@@ -37,6 +37,25 @@ typedef enum {
     ANEURALNETWORKS_TOPK_V2_EX = 50002, /**< Find values and indices of the k largest elements */
     ANEURALNETWORKS_TENSORFLOW_MAX_EX = 50003,  /**< Computes the maximum of elements across dimensions of a tensor */
     ANEURALNETWORKS_SPLIT_EX = 50004, /**< Splits a tensor into sub tensors */
+
+    /**
+     * Computes element-wise reciprocal of square root of the input tensor.
+     *
+     * The output is calculated using this formula:
+     *
+     *      output = 1/sqrt(input)
+     *
+     * Supported tensor {@link OperandCode}:
+     * * {@link ANEURALNETWORKS_TENSOR_FLOAT32}
+     *
+     * Supported tensor rank: up to 4
+     *
+     * Inputs:
+     * * 0: An n-D tensor, specifying the first input.
+     *
+     * Outputs:
+     * * 0: The output tensor, of the same {@link OperandCode} and shape as input0.
+     */
     ANEURALNETWORKS_RSQRT_EX = 50005, /**< Computes reciprocal of square root of x element-wise */
 
     /**