From 952233a1be3b9986bc26086243f37d3a3d8047f2 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Shubham=20Gupta/System=20SW=20/SRI-Bangalore/Engineer/?= =?utf8?q?=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 5 Nov 2018 14:09:08 +0530 Subject: [PATCH] Defining RSQRT_EX in NeuralNetworksEx.h (#3415) This patch defines RSQRT_EX in NeuralNetworksEx.h Signed-off-by: shubham --- include/NeuralNetworksEx.h | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h index a61ca64..3fe9ec8 100644 --- 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 */ /** -- 2.7.4