diff options
Diffstat (limited to 'tensorflow/python/eager/BUILD')
-rw-r--r-- | tensorflow/python/eager/BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/eager/BUILD b/tensorflow/python/eager/BUILD index 6ede8e4f4d..32a8452f62 100644 --- a/tensorflow/python/eager/BUILD +++ b/tensorflow/python/eager/BUILD @@ -322,6 +322,7 @@ cuda_py_test( "//tensorflow/python:math_ops", "//tensorflow/python:pywrap_tensorflow", "//tensorflow/python:random_ops", + "//tensorflow/python/keras", ], ) @@ -404,6 +405,7 @@ cuda_py_test( "//tensorflow/python:array_ops", "//tensorflow/python:client_testlib", "//tensorflow/python:framework_test_lib", + "@six_archive//:six", ], tags = [ "optonly", # The test is too slow in non-opt mode |