此问题与在TFLite转换期间作为不需要的属性的Cast重新排序有关。我们有一个pb文件,其中重整运算符发生在Cast操作之前。我们基本上希望首先在移动GPU中进行重塑操作,该操作将在移动GPU中更快,然后是将在CPU中运行的Cast运算符。但是由于未知原因,通过首先执行Cast运算符,执行流程从GPU中间跳到了CPU本身。这些重塑和Cast转换按此顺序进行,以提高速度。如果将数据从移动GPU复制到移动CPU(如果使用的是较小尺寸而不是4维的尺寸),则证明是确定模型运行时间的一个好因素。即使在pb级别上,Cast也会在Reshape运算符之后显示, tf-lite级别,位置正在改变。当在基准工具和Netron上分析图形时,将引入一个称为Cast-reorder的附加输入。此Cast重新排序是什么,如何消除?
1)在“重塑”级别上剪切图形时,图形符合预期,但没有Cast运算符。
答案 0 :(得分:0)
从描述中还不清楚发生了什么。我需要澄清一下:
在任何一种情况下,可视化都会有所帮助。
还请注意: