Added defenition for CUSTOM unpack ops (#3650)

Defined the constraints for unpack ops

Signed-off-by: Vishal Keshav <vishal.k1@samsung.com>

diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h
index 49e0515..45e6167 100644 (file)
--- a/include/NeuralNetworksEx.h
+++ b/include/NeuralNetworksEx.h
@@ -229,6 +229,27 @@ typedef enum {
     ANEURALNETWORKS_EQUAL_EX = 50011,
     ANEURALNETWORKS_ABS_EX = 50012,
     ANEURALNETWORKS_PACK_EX = 50013,
+    /**
+     * Unpacks a given rank-R tensors into num_splits rank- (R-1) tensors along the axis dimention.
+     * num_splits has to respect integral divisibility of dimention value along axis dimention of the input.
+     *
+     * Supported tensor {@link OperandCode}:
+     * * {@link ANEURALNETWORKS_TENSOR_FLOAT32}
+     * * {@link ANEURALNETWORKS_TENSOR_QUANT8_ASYMM}
+     *
+     * Supported tensor rank: up to 4
+     *
+     * Inputs:
+     * * 0: The input shape is [D0, D1, ..., N at Daxis(i), ..., Dm+1].
+     * * 1: An {@link ANEURALNETWORKS_INT32} scalar, specifying the
+     *      unpack axis.
+     * * 2: An {@link ANEURALNETWORKS_INT32} scalar, specifying the
+     *      number of splits along unpack axis.
+     * Outputs:
+     * * 0 ~ n-1: The list of n output tensors, of shape
+     *            [D0, D1, ..., Daxis(i), ..., Dm]. The output tensors are of the same 
+     *            {@link OperandCode} as the input tensor 0.
+     */
     ANEURALNETWORKS_UNPACK_EX = 50014,
 
     /**