QNN SDK dynamic input shape support or not?

hgwang2016 2025-03-14 11:25:00

从QNN sdk 的converter python源码中,倒是翻到了dynamic shape设定的方法,即在具体维度添加*后缀,但分别在ONNX TF TF-Lite等模型上尝试,均失败。

QNN 什么时候直至动态shape呢?各位有什么好的方法没?

...全文
445 1 打赏 收藏 转发到动态 举报
AI 作业
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
  • 打赏
  • 举报
回复

目前 QNN 不支持开箱即用的动态输入形状。但是,有几种方法可以模拟这种行为:

具有共享权重的多个图。QNN 支持从多个共享模型库创建上下文二进制文件。上下文二进制文件会对权重进行去重,因此您实际上可以在单个二进制文件中调用多个模型/函数。在 qnn 文档中搜索
“context”:{
“weight_sharing_enabled”:True
}
这就是 Llama 或 LLM 通常能够处理提示标记并有效生成标记的方式。

使用切片/掩码将固定形状输入转换为可变形状输入。这可能不一定能减少较短序列的计算量,但可以得到正确的输出。

3,773

社区成员

发帖
与我相关
我的任务
社区描述
本论坛以AI、WoS 、XR、IoT、Auto、生成式AI等核心板块组成,为开发者提供便捷及高效的学习和交流平台。 高通开发者专区主页:https://qualcomm.csdn.net/
人工智能物联网机器学习 技术论坛(原bbs) 北京·东城区
社区管理员
  • csdnsqst0050
  • chipseeker
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧