From 88a2a5c677044c20b3932016fbae6fce0540dc67 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Prasanna=20R/SNAP=20/SRI-Bangalore/Engineer/=EC=82=BC?= =?utf8?q?=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 3 Dec 2018 09:31:35 +0530 Subject: [PATCH] Define SPLIT_EX in NeuralNetworkEx.h (#3755) This patch defines SPLIT_EX in NeuralNetworkEx.h Related issue: #3347 Signed-off-by: prasannar --- include/NeuralNetworksEx.h | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h index f274082..86d9887 100644 --- a/include/NeuralNetworksEx.h +++ b/include/NeuralNetworksEx.h @@ -107,6 +107,27 @@ typedef enum { * * 0: A tensor of the same {@link OperandCode} as input0. */ ANEURALNETWORKS_TENSORFLOW_MAX_EX = 50003, + + /** + * Splits a tensor along a given axis into num_splits subtensors. + * + * Supported tensor {@link OperandCode}: + * * {@link ANEURALNETWORKS_TENSOR_FLOAT32} + * * {@link ANEURALNETWORKS_TENSOR_INT32} + * * {@link ANEURALNETWORKS_TENSOR_QUANT8_ASYMM} + * + * Supported tensor rank: from 1 + * + * Inputs: + * * 0: An n-D tensor to split. + * * 1: An {@link ANEURALNETWORKS_INT32} scalar specifying the axis along + * which to split. + * * 2: An {@link ANEURALNETWORKS_INT32} scalar indicating the number of + * splits along given axis. Must evenly divide axis size. + * + * Outputs: + * * 0 ~ (num_splits - 1): Resulting subtensors. + */ ANEURALNETWORKS_SPLIT_EX = 50004, /**< Splits a tensor into sub tensors */ /** -- 2.7.4