experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Disable mlir support for some tests that fail with it.

Browse Source 
PiperOrigin-RevId: 336385193
Change-Id: I75500fc14cfee384f43a8018e9c9e09654f171b9
This commit is contained in:
Xinyi Wang 2020-10-09 16:17:37 -07:00 committed by TensorFlower Gardener
parent 603b945f8a
commit 8a35913f70
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/keras/distribute/BUILD
View File

@ -335,6 +335,7 @@ distribute_py_test(
distribute_py_test(
name = "distribute_strategy_test",
srcs = ["distribute_strategy_test.py"],
disable_mlir_bridge = True, # TODO(b/170352626)
full_precision = True,
main = "distribute_strategy_test.py",
python_version = "PY3",
@ -402,7 +403,7 @@ distribute_py_test(
name = "keras_dnn_correctness_test",
size = "medium",
srcs = ["keras_dnn_correctness_test.py"],
disable_mlir_bridge = False,
disable_mlir_bridge = True, # TODO(b/170352626)
full_precision = True,
main = "keras_dnn_correctness_test.py",
# Shard count is set to an odd number to distribute tasks across
@ -423,6 +424,7 @@ distribute_py_test(
name = "keras_embedding_model_correctness_test",
size = "medium",
srcs = ["keras_embedding_model_correctness_test.py"],
disable_mlir_bridge = True, # TODO(b/170352626)
full_precision = True,
main = "keras_embedding_model_correctness_test.py",
shard_count = 8,
@ -440,7 +442,7 @@ distribute_py_test(
name = "keras_image_model_correctness_test",
size = "medium",
srcs = ["keras_image_model_correctness_test.py"],
disable_mlir_bridge = False,
disable_mlir_bridge = True, # TODO(b/170352626)
full_precision = True,
main = "keras_image_model_correctness_test.py",
shard_count = 16,
@ -553,6 +555,7 @@ distribute_py_test(
distribute_py_test(
name = "keras_utils_test",
srcs = ["keras_utils_test.py"],
disable_mlir_bridge = True, # TODO(b/170352626)
full_precision = True,
main = "keras_utils_test.py",
shard_count = 4,