diff options
Diffstat (limited to 'tensorflow/contrib/lite/testing/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/testing/BUILD | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/testing/BUILD b/tensorflow/contrib/lite/testing/BUILD index 2edd420fea..3dc666f631 100644 --- a/tensorflow/contrib/lite/testing/BUILD +++ b/tensorflow/contrib/lite/testing/BUILD @@ -20,8 +20,7 @@ load( name = "zip_test_%s" % test_name, size = "large", srcs = ["generated_examples_zip_test.cc"], - args = [ - ] + select({ + args = args + select({ "//tensorflow:android": [], "//conditions:default": [ "--zip_file_path=$(location :zip_%s)" % test_name, @@ -61,7 +60,7 @@ load( "//tensorflow/core:android_tensorflow_test_lib", ], }), -) for conversion_mode, test_name, tags in generated_test_models_all()] +) for conversion_mode, test_name, tags, args in generated_test_models_all()] test_suite( name = "generated_zip_tests", |