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

Break diag_op_test into more shards to prevent timeout.

Browse Source 
PiperOrigin-RevId: 308858659
Change-Id: I5907772bca986e3f21e0b471504fe97af0d45b57
This commit is contained in:
A. Unique TensorFlower 2020-04-28 11:10:40 -07:00 committed by TensorFlower Gardener
parent 35347a5a05
commit cb77658d4e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/kernel_tests/BUILD
View File

@ -1505,7 +1505,7 @@ cuda_py_test(
name = "diag_op_test",
size = "medium",
srcs = ["diag_op_test.py"],
shard_count = 2,
shard_count = 6,
tags = ["no_windows_gpu"],
deps = [
"//tensorflow/python:array_ops",