Adding lite/tools:visualize target into the pip package.

usage:
- python -m tensorflow.lite.tools.visualize INPUT.tflite OUTPUT.html

Tested: Manual
- git clone https://github.com/tensorflow/tensorflow.git --depth=1 tensorflow
- python3 -m venv env
- source env/bin/activate

- pip install numpy wheel
- pip install keras_preprocessing --no-deps

- cd tensorflow
- ./configure

- bazel build -c opt //tensorflow/tools/pip_package:build_pip_package
- ./bazel-bin/tensorflow/tools/pip_package/build_pip_package ../packages

- pip install packages/tensorflow-2.5.0-cp38-cp38-linux_x86_64.whl
- python -m tensorflow.lite.tools.visualize input.tflite output.html

PiperOrigin-RevId: 344189136
Change-Id: If09e06db76f4de1fe37852fd4a0432e78e12a575
This commit is contained in:
Tiezhen WANG 2020-11-24 22:07:43 -08:00 committed by TensorFlower Gardener
parent 6af836f407
commit d844c22f9a

View File

@ -103,6 +103,7 @@ COMMON_PIP_DEPS = [
"//tensorflow/lite/python/testdata:interpreter_test_data",
"//tensorflow/lite/python:tflite_convert",
"//tensorflow/lite/toco/python:toco_from_protos",
"//tensorflow/lite/tools:visualize",
"//tensorflow/python/autograph/core:test_lib",
"//tensorflow/python/autograph/impl/testing:pybind_for_testing",
"//tensorflow/python/autograph/pyct/testing",