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

Tensorflow toco pacakage not found when run with bazel

我正在尝试使用以下命令将 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 运行 命令的正确路径是什么?

在 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