主题中讨论的其他器件:AM69A
工具/软件:
您好:
我正在为设计 ONNX 模型 AM69A 实现 我需要执行一个 Topk 在一个大张量上运行。
我将 RTOS SDK10_01_00_04 与 TIDL Tools 版本 10_01_00_01 搭配使用。
- 输入张量 :
float32[1,1,358400,1] - Attributes:
k=300andaxis=2 The output should bevalues with shape float32[1,1,300,1]indices with shape int64[1,1,300,1]
对于 DSP 实现、我将遵循TIDL_refTopK和中使用的相同方法TIDL_sliceRefProcess。
当我在电路板上运行模型时,它失败了,并且在推理过程中出现错误,我认为它与张量的大小有关:

PS: 运行一个具有输入张量形状的较小模型时所发生的情况 float32[1,1,100,1]和 setK = 30、一切都按预期运行。 模型的执行没有错误,我成功地获得了正确的 TopK 值和索引。 这证实了该逻辑对较小的输入尺寸有效。
您可以在此处找到模型: e2e.ti.com/.../5305.topk_5F00_model.zip
您能帮助我解决我的问题吗?
希望提供任何指导、示例或建议。
此致