tensorflow - 使用 bazel 运行时找不到 Tensorflow toco pacakage

Question

我正在尝试使用以下命令将 pb 文件转换为 tlite 文件：

bazel 运行 //tensorflow/contrib/lite/toco:toco --input_file=sample.pb --output_file=sample.tlite

我收到以下错误：错误：跳过//tensorflow/contrib/lite/toco:toco：在包路径上找不到构建文件。

我在 Python 文件夹下安装了 tensorflow 和 toco。Bazel run 命令的正确路径是什么？

Answer 1

在 python 中安装 toco 不起作用。您必须在 tensorflow 源目录中执行此命令。有关依赖项和配置的必要说明，请参阅https://www.tensorflow.org/install/install_sources。

另请注意，在 bazel args 和命令 args 之间需要一个“--”：

bazel run //tensorflow/contrib/lite/toco:toco -- --input_file=sample.pb --output_file=sample.tlite