Tensorflow Lite GPUdelegate在最后一层给出了“尺寸不是BHWC”错误。但是对我来说，该层似乎是BHWC，就像其他所有层一样

Question

我正在尝试在android上实现YOLO检测器。我遵循了tensorflow提供的演示代码。该模型在CPU或NNAPI上运行良好。但是，当我尝试使用GPU运行它时，程序崩溃，调试器发出以下错误：

    java.lang.IllegalArgumentException: Internal error: Failed to apply delegate: GpuDelegate Prepare: Dimensions are not BHWCNode number 23 (GpuDelegate) failed to prepare.

模型的最后一层看起来像self.conv9 = layers.Conv2D(425, (1,1), strides=(1,1), padding='same', name='conv_9', use_bias=False)

该模型可以在具有CPU或GPU的PC上完美运行。