diff options
author | Nupur Garg <nupurgarg@google.com> | 2018-09-25 14:46:09 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 14:50:12 -0700 |
commit | 20c71535c5f1ed1d918d6cc6e327ffbba49ecbd6 (patch) | |
tree | 16bdfe0e55215f64df518e3fbc1bcf9c01319c42 /tensorflow/contrib/lite/testing/BUILD | |
parent | 038d15d8e2037d4a45e60e076429d67ec7d5ace1 (diff) |
Internal change.
PiperOrigin-RevId: 214507546
Diffstat (limited to 'tensorflow/contrib/lite/testing/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/testing/BUILD | 27 |
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() |