Internal testing change for coverage tests
PiperOrigin-RevId: 313428290 Change-Id: I2a9d6684ddb26e71ea5be4de4c1ed7c1b11cb0d8
This commit is contained in:
parent
b266b46825
commit
7cedc771d4
@ -680,6 +680,9 @@ def gen_model_coverage_test(src, model_name, data, failure_type, tags, size = "m
|
||||
if failure_type[i] != "none":
|
||||
args.append("--failure_type=%s" % failure_type[i])
|
||||
i = i + 1
|
||||
|
||||
# Avoid coverage timeouts for large/enormous tests.
|
||||
coverage_tags = ["nozapfhahn"] if size in ["large", "enormous"] else []
|
||||
native.py_test(
|
||||
name = "model_coverage_test_%s_%s" % (model_name, target_op_sets.lower().replace(",", "_")),
|
||||
srcs = [src],
|
||||
@ -696,7 +699,7 @@ def gen_model_coverage_test(src, model_name, data, failure_type, tags, size = "m
|
||||
"no_gpu", # Executing with TF GPU configurations is redundant.
|
||||
"no_oss",
|
||||
"no_windows",
|
||||
] + tags,
|
||||
] + tags + coverage_tags,
|
||||
deps = [
|
||||
"//tensorflow/lite/testing/model_coverage:model_coverage_lib",
|
||||
"//tensorflow/lite/python:lite",
|
||||
|
Loading…
Reference in New Issue
Block a user