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.

TDA2EVM5777: Issue with softmax layer ?

Part Number: TDA2EVM5777

Hi Kumar,

In the TIDL 0.8.0.0 and 0.1.0.0,  Softmax layer is only supported in the axis 3 i.e width axis only. But, for our use case , We need softmax layer to be applied on axis 1 i.e. channel axis. 

Is this issue solved in the new release of TIDL ?

Also, In the new release TIDeepLearningLibrary_UserGuide document, It is mentioned that softmax layer performs better on DSP than EVE.

Can you guide us How should we run the part of the network on EVE and only softmax layer on DSP ?

Regards,

Sagar

  • Hi Sagar,

    In new release also softmax layer supported for axis 3 only, we are not planning to support softmax for axis 1.

    Use "layersGroupId" to specify to which layers to run on EVE (layersGroupId = 1) and on DSP (layersGroupId = 2). Please refer to VSDK OD(SSD) use case for more details.

    Thanks,
    Praveen