diff options
author | 2018-03-22 17:59:19 -0700 | |
---|---|---|
committer | 2018-03-22 18:02:18 -0700 | |
commit | 80cac4d951d78660b32ccfde9d91ff4050e5d477 (patch) | |
tree | ff176b35b758154b374dbed7122fe42f594f5e84 /tensorflow/contrib/lite/python/BUILD | |
parent | 595723d9db550ba22c187cc7268f05e800d03b67 (diff) |
Convert saved_model to tflite flatbuffer.
PiperOrigin-RevId: 190156133
Diffstat (limited to 'tensorflow/contrib/lite/python/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/python/BUILD | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/python/BUILD b/tensorflow/contrib/lite/python/BUILD index d6f39219ed..ce1a81d06b 100644 --- a/tensorflow/contrib/lite/python/BUILD +++ b/tensorflow/contrib/lite/python/BUILD @@ -84,6 +84,41 @@ py_test( ], ) +py_binary( + name = "convert_savedmodel", + srcs = ["convert_savedmodel.py"], + srcs_version = "PY2AND3", + visibility = ["//visibility:public"], + deps = [ + ":lite", + "//tensorflow/contrib/saved_model:saved_model_py", + "//tensorflow/python:graph_util", + "//tensorflow/python/tools:freeze_graph_lib", + ], +) + +py_test( + name = "convert_savedmodel_test", + srcs = ["convert_savedmodel_test.py"], + srcs_version = "PY2AND3", + visibility = ["//visibility:public"], + deps = [ + ":convert_savedmodel", + "//tensorflow/python:client_testlib", + "//tensorflow/python:platform_test", + "//tensorflow/python:session", + "//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_savedmodel", + ], +) + filegroup( name = "all_files", srcs = glob( |