diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-13 15:42:34 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-13 16:42:50 -0700 |
| commit | 86dd46a3c65c3bcdf92b7ea3ab33bc99ece58236 (patch) | |
| tree | 9230b4e3dbaca45d6481d44a05dd551402142e28 /tensorflow/contrib/lite/python/BUILD | |
| parent | 6a99d23094e953fcd01a86acf2543958fd7bd5bb (diff) | |
Merged commit includes the following changes:
188942386 by mikecase:
Internal Change.
--
188941669 by A. Unique TensorFlower:
Change TensorFlow Lite Java Interpreter to use Checker Framework @NonNull annotation
--
188937484 by zhixianyan:
Convert saved_model to tflite flatbuffer.
--
PiperOrigin-RevId: 188942386
Diffstat (limited to 'tensorflow/contrib/lite/python/BUILD')
| -rw-r--r-- | tensorflow/contrib/lite/python/BUILD | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/python/BUILD b/tensorflow/contrib/lite/python/BUILD index 76607af079..97bcb8eec2 100644 --- a/tensorflow/contrib/lite/python/BUILD +++ b/tensorflow/contrib/lite/python/BUILD @@ -81,6 +81,35 @@ 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/debug:local_cli_wrapper", + "//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:dtypes", + "//tensorflow/python:platform_test", + "//tensorflow/python:session", + "//tensorflow/python/saved_model", + ], +) + filegroup( name = "all_files", srcs = glob( |