aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/pip_package
diff options
context:
space:
mode:
authorGravatar Andrew Selle <aselle@google.com>2018-08-20 16:29:00 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-20 16:32:31 -0700
commit631e86bf2d07ee6a530f552c8e7a484e1a28a0e1 (patch)
treee8e2c88270265c9e3d7b130e4cdd868201d5eb47 /tensorflow/tools/pip_package
parentfc4504edb1ab419ae59b0ebb9ff8d943beb61117 (diff)
Enable TOCO and TFLite In Windows.
- Include in pip build command - Fix use of subprocess to not use unsupported flag on windows. - Fix __init__ in contrib to include tflite PiperOrigin-RevId: 209505929
Diffstat (limited to 'tensorflow/tools/pip_package')
-rw-r--r--tensorflow/tools/pip_package/BUILD7
-rw-r--r--tensorflow/tools/pip_package/MANIFEST.in1
2 files changed, 7 insertions, 1 deletions
diff --git a/tensorflow/tools/pip_package/BUILD b/tensorflow/tools/pip_package/BUILD
index 00c1337b19..7839eddcf8 100644
--- a/tensorflow/tools/pip_package/BUILD
+++ b/tensorflow/tools/pip_package/BUILD
@@ -207,7 +207,12 @@ sh_binary(
name = "build_pip_package",
srcs = ["build_pip_package.sh"],
data = select({
- "//tensorflow:windows": [":simple_console_for_windows"],
+ "//tensorflow:windows": [
+ ":simple_console_for_windows",
+ "//tensorflow/contrib/lite/python:interpreter_test_data",
+ "//tensorflow/contrib/lite/python:tflite_convert",
+ "//tensorflow/contrib/lite/toco/python:toco_from_protos",
+ ],
"//conditions:default": COMMON_PIP_DEPS + [
":simple_console",
"//tensorflow/contrib/lite/python:interpreter_test_data",
diff --git a/tensorflow/tools/pip_package/MANIFEST.in b/tensorflow/tools/pip_package/MANIFEST.in
index 86c5e4776d..c4b4af93b8 100644
--- a/tensorflow/tools/pip_package/MANIFEST.in
+++ b/tensorflow/tools/pip_package/MANIFEST.in
@@ -1,5 +1,6 @@
include README
recursive-include * *.py
+recursive-include * *.pd
recursive-include * *.so
recursive-include * *.dll
recursive-include * *.lib