DepthToSpace运行在cpu上

casiatao · 2022 年8 月 15 日 03:17

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

颜值即正义 · 2022 年8 月 15 日 04:38

你好，DepthToSpace算子为被动量化，想让该算子跑在bpu上，在该模型前后插入一个不改变数据的unitconv即可。

casiatao · 2022 年8 月 15 日 12:31

好的，谢谢！

颜值即正义 · 2022 年8 月 15 日 13:02

不客气，加油加油