TIDL Argmax Layer Question

Ahan Tseng

Prodigy 190 points

Hi,

I have a model wtih argmax layer. And the output dimension of the argmax layer is 1x7x10x10.

When I run the model on DSP and EVE mode, I get two difference results.

On the DSP mode, the max output is 5.

On the EVE mode, the max output is 6.

And on the TIDL, tidl_argmax.h, I found the definition of the max channels.

#ifdef CORE_EVE
#define ARGMAX_MAX_NUM_CHANNELS (15)
#elif CORE_DSP
#define ARGMAX_MAX_NUM_CHANNELS (6)
#endif

So, how can I resolve the question ?

Thanks and Regards,

Ahan

over 8 years ago

0 kumar.desappan over 8 years ago

TI__Mastermind 22145 points

Hi Ahan,

The Argmax layer data flow needs to be changed to handle more than 6 channels, Currently we process all the channels together, we may have modify the flow to process one channel at a time and have the max index in the internal memory between the channel processing.

Thanks and Rgerads,

Kumar.D

0 Ahan Tseng over 8 years ago in reply to kumar.desappan

Prodigy 190 points

Hi Kumar,

Is there a solution to this in the future?

Thanks and Regards,

Ahan

0 kumar.desappan over 8 years ago in reply to Ahan Tseng

TI__Mastermind 22145 points

Ahan,

As of now it is not in the plan. We are working on plan for next release. Will update here after finalizing the plan.

Thanks and Regards,

Kumar.D

Processors

Processors forum

TIDL Argmax Layer Question