×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

模型移植时min,max,addmm算子不支持问题 已完结 sunxiaofeng2022-07-21 16:52:03 回复 5 查看 使用求助
模型移植时min,max,addmm算子不支持问题
分享到:

在DQN网络移植过程中,使用寒武纪pytorch,模型已经量化,在进行推理时遇到如下warning816610de8703a78721c91fc0be4611b.png

到所提示的op_methods.cpp文件下进行查看,发现有该3个算子,但是实现的逻辑都是先到cpu实现后把结果搬到mlu上,请问如何移除这个warning。

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭