Softmax layer is only supported in the axis 3. But, for my use case , We need softmax layer to be applied on axis 1. Can I use C66 to do the softmax?
This thread has been locked.
If you have a related question, please click the "Ask a related question" button in the top right corner. The newly created question will be automatically linked to this question.
Softmax layer is only supported in the axis 3. But, for my use case , We need softmax layer to be applied on axis 1. Can I use C66 to do the softmax?
If this layer is used only at the end of the network as post-processing, this can be implemented in either ARM, C6x outside TIDL