diff options
author | Nupur Garg <nupurgarg@google.com> | 2018-05-30 17:54:02 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-30 17:56:47 -0700 |
commit | 316549d36f6ab3d250ce9e33b768bbfb1a4d7362 (patch) | |
tree | cef32a4c8ace3dedac532c14fd39944d5bc4ed2b /tensorflow/contrib/lite/python/BUILD | |
parent | 2a484497062677f5cf0205ee3b9c28a64f03fe04 (diff) |
Enable TOCO pip command line binding.
PiperOrigin-RevId: 198649827
Diffstat (limited to 'tensorflow/contrib/lite/python/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/python/BUILD | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/tensorflow/contrib/lite/python/BUILD b/tensorflow/contrib/lite/python/BUILD index a40e512045..7e6ff6c0a8 100644 --- a/tensorflow/contrib/lite/python/BUILD +++ b/tensorflow/contrib/lite/python/BUILD @@ -36,6 +36,16 @@ py_test( ], ) +py_binary( + name = "tflite_convert", + srcs = ["tflite_convert.py"], + srcs_version = "PY2AND3", + visibility = ["//visibility:public"], + deps = [ + ":lite", + ], +) + py_library( name = "lite", srcs = ["lite.py"], @@ -125,6 +135,7 @@ py_library( srcs_version = "PY2AND3", visibility = ["//visibility:public"], deps = [ + ":convert", "//tensorflow/contrib/saved_model:saved_model_py", "//tensorflow/python:graph_util", "//tensorflow/python:platform", @@ -164,11 +175,3 @@ py_test( "//tensorflow/python/saved_model", ], ) - -# Transitive dependencies of this target will be included in the pip package. -py_library( - name = "tf_lite_py_pip", - deps = [ - ":convert_saved_model", - ], -) |