CMSIS-NN: Add softmax s16 support (#1425)
CMSIS-NN: Add softmax s16 support
Adds scalar version of softmax with int8 input and int16 output.
Adds scalar version of softmax with int16 input and int16 output.
Adds unit tests.
diff --git a/ARM.CMSIS.pdsc b/ARM.CMSIS.pdsc
index 4fd9e8f..cee05b7 100644
--- a/ARM.CMSIS.pdsc
+++ b/ARM.CMSIS.pdsc
@@ -21,6 +21,7 @@
- Support for int16 depthwise conv for reference implementation including dilation
- Support for int16 average and max pooling for reference implementation
- Support for elementwise add and mul int16 scalar version
+ - Support for softmax int16 scalar version
CMSIS-RTOS2:
- RTX 5.5.4 (see revision history for details)
</release>
@@ -2283,6 +2284,8 @@
<file category="source" name="CMSIS/NN/Source/FullyConnectedFunctions/arm_fully_connected_mat_q7_vec_q15.c"/>
<file category="source" name="CMSIS/NN/Source/SoftmaxFunctions/arm_softmax_q15.c"/>
<file category="source" name="CMSIS/NN/Source/SoftmaxFunctions/arm_softmax_s8.c"/>
+ <file category="source" name="CMSIS/NN/Source/SoftmaxFunctions/arm_softmax_s8_s16.c"/>
+ <file category="source" name="CMSIS/NN/Source/SoftmaxFunctions/arm_softmax_s16.c"/>
<file category="source" name="CMSIS/NN/Source/SoftmaxFunctions/arm_softmax_u8.c"/>
<file category="source" name="CMSIS/NN/Source/SoftmaxFunctions/arm_softmax_q7.c"/>
<file category="source" name="CMSIS/NN/Source/SoftmaxFunctions/arm_softmax_with_batch_q7.c"/>