diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-13 10:22:53 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-13 10:27:03 -0700 |
commit | 29c129c69d3aa602d0fa083f7477144466f562d1 (patch) | |
tree | 595d8ae3f5beecc62bfb81bc2ad3202524d0e145 /tensorflow/contrib/lite/python/BUILD | |
parent | fce49887e827abc2627fd2a7bc135800baaafc4f (diff) |
Add TFLite Interpreter bindings.
PiperOrigin-RevId: 188890943
Diffstat (limited to 'tensorflow/contrib/lite/python/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/python/BUILD | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/python/BUILD b/tensorflow/contrib/lite/python/BUILD index 82feae0f00..76607af079 100644 --- a/tensorflow/contrib/lite/python/BUILD +++ b/tensorflow/contrib/lite/python/BUILD @@ -4,6 +4,38 @@ package(default_visibility = ["//tensorflow:internal"]) load("//tensorflow:tensorflow.bzl", "py_test") +filegroup( + name = "interpreter_test_data", + srcs = glob(["**/testdata/*"]), + visibility = ["//tensorflow:__subpackages__"], +) + +py_library( + name = "interpreter", + srcs = [ + "interpreter.py", + ], + srcs_version = "PY2AND3", + visibility = ["//visibility:public"], + deps = [ + "//tensorflow/contrib/lite/python/interpreter_wrapper:tensorflow_wrap_interpreter_wrapper", + ], +) + +py_test( + name = "interpreter_test", + srcs = ["interpreter_test.py"], + data = [":interpreter_test_data"], + srcs_version = "PY2AND3", + tags = ["no_oss"], + deps = [ + ":interpreter", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:platform_test", + ], +) + py_library( name = "lite", srcs = ["lite.py"], |