diff options
Diffstat (limited to 'tensorflow/contrib/lite/testing/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/testing/BUILD | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/testing/BUILD b/tensorflow/contrib/lite/testing/BUILD index a4736bfee9..55ef1172b2 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,32 @@ cc_binary( ], ) +py_binary( + name = "model_coverage_lib", + srcs = ["//tensorflow/contrib/lite/testing:model_coverage/model_coverage_lib.py"], + srcs_version = "PY2AND3", + tags = ["no_pip"], + 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_pip", + "no_windows", + "notap", + ], + deps = [ + ":model_coverage_lib", + "//tensorflow/python:client_testlib", + ], +) + tflite_portable_test_suite() |