From 07e504917146d7ca5c5c7f055699254b6a26ce99 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=B5=9C=ED=98=95=EA=B7=9C/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Thu, 31 May 2018 16:56:29 +0900 Subject: [PATCH] Update comment for STRIDED_SLICE in NeuralNetworks.h (#1460) - Update comment for STRIDED_SLICE in NeuralNetworks.h, because operands of STRIDED_SLICE has been implemented by following parameters defined in the latest Android NN API. Signed-off-by: Hyung-Kyu Choi --- include/NeuralNetworks.h | 40 +++++++++++++++++++++++++++------------- 1 file changed, 27 insertions(+), 13 deletions(-) diff --git a/include/NeuralNetworks.h b/include/NeuralNetworks.h index 5f481c0..11afb53 100644 --- a/include/NeuralNetworks.h +++ b/include/NeuralNetworks.h @@ -1201,28 +1201,42 @@ typedef enum { /** * Extracts a strided slice of a tensor. * - * Roughly speaking, this op extracts a slice of size (end - begin) / stride from the given - * input tensor. Starting at the location specified by begin the slice continues by adding - * stride to the index until all dimensions are not less than end. Note that a stride can - * be negative, which causes a reverse slice. + * Roughly speaking, this op extracts a slice of size (end - begin) / stride + * from the given input tensor. Starting at the location specified by begin + * the slice continues by adding stride to the index until all dimensions + * are not less than end. Note that a stride can be negative, which causes a + * reverse slice. * - * Supported tensor types: + * Supported tensor {@link OperandCode}: * * {@link ANEURALNETWORKS_TENSOR_FLOAT32} * * {@link ANEURALNETWORKS_TENSOR_QUANT8_ASYMM} * * Supported tensor rank: up to 4 * * Inputs: - * 0: An n-D tensor, specifying the tensor to be sliced. - * 1: A 1-D Tensor of type TENSOR_INT32, the starts of the dimensions of the input - * tensor to be sliced. The length must be of rank(input0). - * 2: A 1-D Tensor of type TENSOR_INT32, the ends of the dimensions of the input - * tensor to be sliced. The length must be of rank(input0). - * 3: A 1-D Tensor of type TENSOR_INT32, the strides of the dimensions of the input - * tensor to be sliced. The length must be of rank(input0). + * * 0: An n-D tensor, specifying the tensor to be sliced. + * * 1: A 1-D Tensor of {@link ANEURALNETWORKS_TENSOR_INT32}, the starts of + * the dimensions of the input tensor to be sliced. The length must be + * of rank(input0). + * * 2: A 1-D Tensor of {@link ANEURALNETWORKS_TENSOR_INT32}, the ends of + * the dimensions of the input tensor to be sliced. The length must be + * of rank(input0). + * * 3: A 1-D Tensor of {@link ANEURALNETWORKS_TENSOR_INT32}, the strides of + * the dimensions of the input tensor to be sliced. The length must be + * of rank(input0). + * * 4: An {@link ANEURALNETWORKS_INT32} scalar, begin_mask. If the ith bit + * of begin_mask is set, begin[i] is ignored and the fullest possible + * range in that dimension is used instead. + * * 5: An {@link ANEURALNETWORKS_INT32} scalar, end_mask. If the ith bit of + * end_mask is set, end[i] is ignored and the fullest possible range in + * that dimension is used instead. + * * 6: An {@link ANEURALNETWORKS_INT32} scalar, shrink_axis_mask. An int32 + * mask. If the ith bit of shrink_axis_mask is set, it implies that the + * ith specification shrinks the dimensionality by 1. A slice of size 1 + * starting from begin[i] in the dimension must be preserved. * * Outputs: - * 0: A tensor of the same type as input0. + * * 0: A tensor of the same {@link OperandCode} as input0. */ ANEURALNETWORKS_STRIDED_SLICE = 35, -- 2.7.4