了解tf2onnx解析器中的lstm_cell模式

时间:2018-10-28 15:10:29

标签: tensorflow onnx

我试图转换在https://github.com/onnx/tensorflow-onnx/tree/master/tests/models/lstm中找到的lstm Frozen.pb

但是tf2onnx解析器无法将其识别为lstm模型,而是将其视为普通模型。当我尝试调试时,我发现此OpTypePattern位于 https://github.com/onnx/tensorflow-onnx/blob/master/tf2onnx/rewriter/rnn_utils.py#L98不匹配,因此无法识别为lstm模型

有人知道这种模式是如何形成的以及在什么基础上形成的吗?预先感谢。

0 个答案:

没有答案