diff options
| author | Shubham Gupta/System SW /SRI-Bangalore/Engineer/삼성전자 <shub98.gupta@samsung.com> | 2018-11-05 14:09:08 +0530 |
|---|---|---|
| committer | 오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com> | 2018-11-05 17:39:08 +0900 |
| commit | 952233a1be3b9986bc26086243f37d3a3d8047f2 (patch) | |
| tree | d056c9ed31e468b4fa6b76752974ff4dbe62074f /include | |
| parent | 2bb67ff67e030583d959354ffb60adfdb807e3c3 (diff) | |
| download | nnfw-952233a1be3b9986bc26086243f37d3a3d8047f2.tar.gz nnfw-952233a1be3b9986bc26086243f37d3a3d8047f2.tar.bz2 nnfw-952233a1be3b9986bc26086243f37d3a3d8047f2.zip | |
Defining RSQRT_EX in NeuralNetworksEx.h (#3415)
This patch defines RSQRT_EX in NeuralNetworksEx.h
Signed-off-by: shubham <shub98.gupta@samsung.com>
Diffstat (limited to 'include')
| -rw-r--r-- | include/NeuralNetworksEx.h | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h index a61ca644a..3fe9ec84e 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 */ /** |