×

签到

分享到微信

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

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

tfv2 加法计算速度咨询 已解决 lichaoqian2025-06-03 14:38:14 回复 3 查看 技术答疑 使用求助 经验交流
tfv2 加法计算速度咨询
分享到:

Tensorflow里面AddV2性能问题

Tensorflow在计算half加法时两个Tensor[1,100,1,152,1,256], [1,100,2,152,2,256]相加需要时间接近10ms。
在计算小尺寸的时候,比如[1,50,1,76,1,256], [1,50,2,76,2,256]相加,没有问题,时间约为0.15ms。

环境为cambricon- /tensorflow:v25.01.0-tensorflow2.12-v1.28.1-ubuntu22.04-py310
370


测试代码和profile结果在附件中。

attachment.zip


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