请注意,本文内容源自机器翻译,可能存在语法或其它翻译错误,仅供参考。如需获取准确内容,请参阅链接中的英语原文或自行翻译。
器件型号:AM69A工具/软件:
TI 团队大家好、
我正在实施自定义运算符、并且在上运行推理时遇到可重现的问题 来实现 。
问题总结:
- 。 自定义 op 输出张量 具有 4D 形状(例如)
[1, 4, 3, 4]。 - 当我为
invoke函数内部的输出(在 DSP 上)分配值时、i 在分配后立即打印值 它们出现了 100%正确 。 - 然而,运算符运行后,模型的最终输出(例如,我从推理结果获得的内容)包含 值不正确或损坏 —它们与我在分配过程中打印的内容不匹配。
- 。 相同的定制操作码 、形状相同、运行 完美地放置在 PC 上 如何使用 TIDL 推理。
其他意见:
- 为了隔离问题,我从自定义操作中删除所有逻辑,然后用常量值填充输出(例如,)
output[i] = 10。 即使在这种情况下、DSP 输出仍然不正确。 - 出现问题 仅当输出张量有 4 个维度时 和时 通道尺寸(第二或第三索引) 大于 2 时才有意义。
- 这表明问题是这样的 但它不在计算中 、但可能是 DSP 运行时处理多维输出缓冲区的方式。
问题:
- 在自定义 DSP 操作中写入 4D 输出张量是否有任何已知问题或限制?
非常感谢您对调试该问题的支持。
此致、