diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/NeuralNetworksEx.h | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/include/NeuralNetworksEx.h b/include/NeuralNetworksEx.h index d8c39e9e0..2c8fd9dc8 100644 --- a/include/NeuralNetworksEx.h +++ b/include/NeuralNetworksEx.h @@ -192,6 +192,30 @@ typedef enum { * * 0: A tensor of the same {@link OperandCode} as input0. */ ANEURALNETWORKS_TENSORFLOW_SUM_EX = 50009, + + /** + * A transposed convolutional layer carries out a regular convolution but reverts its spatisal transformation. + * Transpose convolution basically performs convolution with transposed weights. + * + * Supported tensor {@link OperandCode}: + * {@link ANEURALNETWORKS_TENSOR_FLOAT32} + * {@link ANEURALNETWORKS_TENSOR_INT32} + * + * Supported tensor rank: only 4 + * + * Inputs: + * 0: An {@link ANEURALNETWORKS_INT32} 1-D four element tensor, specifying the output shape. + * 1: A 4-D tensor, of shape [depth_out, filter_height, filter_width, depth_in], specifying the filter. + * 2: A 4-D tensor, of shape [batches, height, width, depth_in], specifying the input. + * 3: An {@link ANEURALNETWORKS_INT32} scalar, specifying the padding type. + * 4: An {@link ANEURALNETWORKS_INT32} scalar, specifying the stride when + * walking through input in the ‘width’ dimension. + * 5: An {@link ANEURALNETWORKS_INT32} scalar, specifying the stride when + * walking through input in the height dimension. + * + * Outputs: + * 0: The output 4-D tensor, of shape [batches, out_height, out_width, depth_out]. + */ ANEURALNETWORKS_TRANSPOSE_CONV_EX = 50010, /** |