Increase size of stack_ops_test to medium. It previously timed out occasionally.

PiperOrigin-RevId: 340701018
Change-Id: I9b28510ee806088a0466a42d6cca00d19441ea57
This commit is contained in:
Jeremy Lau 2020-11-04 12:04:46 -08:00 committed by TensorFlower Gardener
parent 20e15fa0f5
commit 5adba3f435

View File

@ -85,11 +85,8 @@ cuda_py_test(
cuda_py_test(
name = "stack_op_test",
size = "small",
size = "medium",
srcs = ["stack_op_test.py"],
xla_tags = [
"no_cuda_asan", # times out
],
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",