专栏算法工具链DepthToSpace运行在cpu上

DepthToSpace运行在cpu上

已解决
casiatao2022-08-15
33
3

我的DepthToSpace算子运行在CPU上,但是输入输出是满足要求的,请问为什么会这样

算法工具链
评论1
0/1000
  • 颜值即正义
    Lv.2
    你好,DepthToSpace算子为被动量化,想让该算子跑在bpu上,在该模型前后插入一个不改变数据的unitconv即可。
    2022-08-15
    0
    2
    • casiatao回复颜值即正义:

      好的,谢谢!

      2022-08-15
      0
    • 颜值即正义回复casiatao:

      不客气,加油加油

      2022-08-15
      3