diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2017-02-17 17:05:49 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-17 17:23:48 -0800 |
| commit | 93a975e114ee1c35f01ed3bdd47170e6f7129014 (patch) | |
| tree | e34255aff698fe6a4a586e7940337fd278947f58 /tensorflow/python/BUILD | |
| parent | eb9624017a0040e805fda622a5f9ec6681e24246 (diff) | |
Merge changes from github.
Change: 147897309
Diffstat (limited to 'tensorflow/python/BUILD')
| -rw-r--r-- | tensorflow/python/BUILD | 36 |
1 files changed, 35 insertions, 1 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 5353035b18..04e1afaf81 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -2636,7 +2636,6 @@ cuda_py_tests( "training/proximal_gradient_descent_test.py", "training/queue_runner_test.py", "training/rmsprop_test.py", - "training/saver_test.py", "training/slot_creator_test.py", "training/tensorboard_logging_test.py", "training/training_ops_test.py", @@ -2678,6 +2677,41 @@ cuda_py_tests( ], ) +cuda_py_test( + name = "saver_test", + size = "medium", + srcs = [ + "training/saver_test.py", + ], + additional_deps = [ + ":array_ops", + ":client_testlib", + ":control_flow_ops", + ":data_flow_ops", + ":data_flow_ops_gen", + ":errors", + ":gradients", + ":math_ops", + ":nn_grad", + ":nn_ops", + ":partitioned_variables", + ":platform", + ":platform_test", + ":pywrap_tensorflow", + ":random_ops", + ":resource_variable_ops", + ":sparse_ops", + ":summary", + ":training", + ":util", + ":variable_scope", + ":variables", + "//third_party/py/numpy", + "@six_archive//:six", + "//tensorflow/core:protos_all_py", + ], +) + py_test( name = "saver_large_variable_test", size = "small", |