tensorrt - tensorRT 和 tf.while_loop

Question

我正在将自定义模型转换为包含“while_loop”的 TensorRT。'while_loop' 迭代固定数量的步骤。

这篇 GIT 帖子指出 TensorRT 与 while_loop 存在问题： https ://github.com/tensorflow/tensorrt/issues/87

我得到的当前错误是。此错误对应于我在 while 循环中创建的临时变量。

uff.model.exceptions.UffException: Const node conversion requested, but node is not Const

TensorRT 是否支持 while_loops？我的代码是在 TF 1.14 中实现的。如果不是，TensorRT 是否支持在 TF 2.0 中实现的 for 循环？

如果我删除 while_loop，我可以创建一个 UFF 模型。

提前致谢。