diff options
| author |  Nupur Garg <nupurgarg@google.com> | 2018-09-26 11:18:09 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-26 11:23:05 -0700 |
| commit | cccace731712ab192448ff9bc3ce7259fd462e74 (patch) | |
| tree | b6b9a6b361ff38b788f05e2aecdb21f524570767 /tensorflow/contrib/lite/testing/BUILD | |
| parent | 5cedb0427bd4db4117182da8bc0680dd555b4f49 (diff) | |
Internal change.
PiperOrigin-RevId: 214636032
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() |