aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/testing/BUILD
diff options
context:
space:
mode:
authorGravatar Nupur Garg <nupurgarg@google.com>2018-09-25 14:46:09 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-25 14:50:12 -0700
commit20c71535c5f1ed1d918d6cc6e327ffbba49ecbd6 (patch)
tree16bdfe0e55215f64df518e3fbc1bcf9c01319c42 /tensorflow/contrib/lite/testing/BUILD
parent038d15d8e2037d4a45e60e076429d67ec7d5ace1 (diff)
Internal change.
PiperOrigin-RevId: 214507546
Diffstat (limited to 'tensorflow/contrib/lite/testing/BUILD')
-rw-r--r--tensorflow/contrib/lite/testing/BUILD27
1 files changed, 27 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/testing/BUILD b/tensorflow/contrib/lite/testing/BUILD
index a4736bfee9..c4a2b03444 100644
--- a/tensorflow/contrib/lite/testing/BUILD
+++ b/tensorflow/contrib/lite/testing/BUILD
@@ -13,6 +13,7 @@ load("//tensorflow/contrib/lite:special_rules.bzl", "tflite_portable_test_suite"
load(
"//tensorflow:tensorflow.bzl",
"tf_cc_test",
+ "py_test",
)
[gen_zip_test(
@@ -362,4 +363,30 @@ cc_binary(
],
)
+py_binary(
+ name = "model_coverage_lib",
+ srcs = ["//tensorflow/contrib/lite/testing:model_coverage/model_coverage_lib.py"],
+ srcs_version = "PY2AND3",
+ visibility = ["//tensorflow/contrib/lite:__subpackages__"],
+ deps = [
+ "//tensorflow/contrib/lite/python:lite",
+ "//tensorflow/python:platform",
+ ],
+)
+
+py_test(
+ name = "model_coverage_lib_test",
+ srcs = ["//tensorflow/contrib/lite/testing:model_coverage/model_coverage_lib_test.py"],
+ srcs_version = "PY2AND3",
+ tags = [
+ "no_oss",
+ "no_windows",
+ "notap",
+ ],
+ deps = [
+ ":model_coverage_lib",
+ "//tensorflow/python:client_testlib",
+ ],
+)
+
tflite_portable_test_suite()