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

Disable keras layer_correctness_test for TFRT.

Browse Source 
PiperOrigin-RevId: 357002490
Change-Id: I28966d39de769c7388a0ef300d2006c070b20c14
This commit is contained in:
Chuanhao Zhuge 2021-02-11 10:14:29 -08:00 committed by TensorFlower Gardener
parent 2d7463e97e
commit d73cfcdbba
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/keras/mixed_precision/BUILD
View File

@ -289,7 +289,10 @@ cuda_py_test(
srcs = ["layer_correctness_test.py"], srcs = ["layer_correctness_test.py"],
python_version = "PY3", python_version = "PY3",
shard_count = 10, shard_count = 10,
tags = ["no_rocm"], tags = [
"no_rocm",
"no_tfrt", # TODO(b/179863362)
],
deps = [ deps = [
"//tensorflow/python:client_testlib", "//tensorflow/python:client_testlib",
"//tensorflow/python/compat:v2_compat", "//tensorflow/python/compat:v2_compat",