无法将模型转换为 tensorflow 的精简版格式

Question

我正在尝试在移动设备上使用训练有素的 rnn。问题是，当我使用 toco 将我的 .pb 文件转换为 .tflite 时，它失败并显示以下错误消息：

WARNING: Config values are not defined in any .rc file: opt.
INFO: Found 1 target...
Target //tensorflow/contrib/lite/toco:toco up-to-date:
  bazel-bin/tensorflow/contrib/lite/toco/toco
INFO: Elapsed time: 0.287s, Critical Path: 0.00s

INFO: Running command line: bazel-bin/tensorflow/contrib/lite/toco/toco '--input_file=/Users/valentinradu/Playgrounds/char-rnn-tensorflow/remote_save/latest/graph_frz.pb' '--output_file=/Users/valentinradu/Playgrounds/char-rnn-tensorflow/remote_save/latest/graph.tflite' '--input_format=TENSORFLOW_GRAPHDEF' '--output_format=TFLITE' '--input_type=FLOAT' '--inference_type=FLOAT' '--input_shapes=1,128:1,50,50' '--input_arrays=state_in,data_in' '--output_arrays=state_out,data_out'
2017-11-16 06:48:00.156091: I tensorflow/contrib/lite/toco/import_tensorflow.cc:937] Converting unsupported operation: Fill
2017-11-16 06:48:00.156811: I tensorflow/contrib/lite/toco/import_tensorflow.cc:937] Converting unsupported operation: Fill
2017-11-16 06:48:00.156821: I tensorflow/contrib/lite/toco/import_tensorflow.cc:937] Converting unsupported operation: Pack
2017-11-16 06:48:00.156829: I tensorflow/contrib/lite/toco/import_tensorflow.cc:937] Converting unsupported operation: Pack
2017-11-16 06:48:00.156841: I tensorflow/contrib/lite/toco/import_tensorflow.cc:937] Converting unsupported operation: Unpack
2017-11-16 06:48:00.156856: I tensorflow/contrib/lite/toco/import_tensorflow.cc:937] Converting unsupported operation: StridedSlice
2017-11-16 06:48:00.156872: I tensorflow/contrib/lite/toco/import_tensorflow.cc:937] Converting unsupported operation: StridedSlice
2017-11-16 06:48:00.157260: I tensorflow/contrib/lite/toco/import_tensorflow.cc:937] Converting unsupported operation: Pack
2017-11-16 06:48:00.157277: I tensorflow/contrib/lite/toco/import_tensorflow.cc:937] Converting unsupported operation: Pack
2017-11-16 06:48:00.158053: I tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39] Before general graph transformations: 40 operators, 64 arrays (0 quantized)
2017-11-16 06:48:00.158141: F tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc:793] Check failed: indices_shape.dimensions_count() == 1 (2 vs. 1)

现在，我明白这是因为 toco 期望我的模型只有一个输入，但是，在我发现的示例中，到目前为止人们使用 2 个输入没有问题。

这是否与我在模型中使用的操作有关？这是已知限制还是错误？

Here 是我构建模型的方式。

Answer 1

不幸的是，这似乎是收集运算符的限制。我会把它留在这里，以防其他人将来遇到这个问题。

如果任何在 tensorflow 方面或团队中有更多经验的人可以证实这一点，那就太好了！

Answer 2

tf.gather 接受一个 indices 参数，这是一个一维整数数组。

你得到的错误是说在你的图表中，Gather op 有一个 indices 数组，它是二维的而不是预期的一维数组。

更新：我不知道，但在 Gather 中拥有多于一维的索引数组实际上是合法的。将其视为 TFLite 转换器 (toco) 中的 "not implemented yet" 错误。

无法将模型转换为 tensorflow 的精简版格式

Can't convert model to tensorflow's lite format

tensorflow

tensorflow-lite