如何在GPU上运行`FusedResizeAndPadConv2D'（或配置`optimize_for_inference`不要融合`resize`和`conv2d`）？

Question

我的图表在resize操作之前有conv2d操作，我在调用optimize_for_inference后发现，这两个操作符被融合为FusedResizeAndPadConv2D操作符。我认为这是一个优化。但是，当我推断它时，我发现生成的图形非常慢。在进行一些分析之后，我发现FusedResizeAndPadConv2D在CPU上进行了评估（而在我没有resize的原始图表中，conv2d是用GPU评估的。

所以我想知道，如何在GPU上运行FusedResizeAndPadConv2D。或者，如何配置optimize_for_inference以不融合我的resize和conv2d？

Answer 1

根据此doc，仅为FusedResizeAndPadConv2D生成ResizeBilinear。使用NEAREST_NEIGHBOR作为tf.image.resize_images的调整大小方法解决了我的问题。