aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/python/BUILD
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-03-13 10:22:53 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-03-13 10:27:03 -0700
commit29c129c69d3aa602d0fa083f7477144466f562d1 (patch)
tree595d8ae3f5beecc62bfb81bc2ad3202524d0e145 /tensorflow/contrib/lite/python/BUILD
parentfce49887e827abc2627fd2a7bc135800baaafc4f (diff)
Add TFLite Interpreter bindings.
PiperOrigin-RevId: 188890943
Diffstat (limited to 'tensorflow/contrib/lite/python/BUILD')
-rw-r--r--tensorflow/contrib/lite/python/BUILD32
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"],