在 tflite 模型中查找动态张量

Finding dynamic tensors in a tflite model

我目前在使用 C API 加载 tflite 模型时遇到以下错误:

ERROR: Attempting to use a delegate that only supports static-sized tensors with a graph that has dynamic-sized tensors.

可以找到tflite模型here. It is a tflite conversion of the LEAF model

经检查输入和输出张量似乎具有静态大小。我用 Netron 检查了模型,但找不到任何动态张量,但我可能忽略了。有没有办法查看哪些张量具体导致其动态张量出现问题?

尽管图中没有动态大小的张量,但上图有一个控制流操作,While 操作。目前,具有控制流操作的图被视为动态图,硬件加速委托不支持这些图,硬件加速委托只允许静态图结构。

在您的 gradle 文件中添加此依赖项以支持 ops 模型:

org.tensorflow:tensorflow-lite-select-tf-ops:2.8.0