From 4d7d62d5101f069017c8714c53299be022b4ff74 Mon Sep 17 00:00:00 2001 From: Anjali Sridhar Date: Tue, 20 Mar 2018 14:23:39 -0700 Subject: [PATCH] Internal Change. PiperOrigin-RevId: 189809845 --- tensorflow/python/framework/test_util.py | 6 +++--- tensorflow/python/keras/BUILD | 14 ++++++++------ 2 files changed, 11 insertions(+), 9 deletions(-) diff --git a/tensorflow/python/framework/test_util.py b/tensorflow/python/framework/test_util.py index e9e86e4..d8f8569 100644 --- a/tensorflow/python/framework/test_util.py +++ b/tensorflow/python/framework/test_util.py @@ -902,9 +902,9 @@ class TensorFlowTestCase(googletest.TestCase): Use the `use_gpu` and `force_gpu` options to control where ops are run. If `force_gpu` is True, all ops are pinned to `/device:GPU:0`. Otherwise, if - `use_gpu` - is True, TensorFlow tries to run as many ops on the GPU as possible. If both - `force_gpu and `use_gpu` are False, all ops are pinned to the CPU. + `use_gpu` is True, TensorFlow tries to run as many ops on the GPU as + possible. If both `force_gpu and `use_gpu` are False, all ops are pinned to + the CPU. Example: ```python diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD index eef91e9..3180b9f 100755 --- a/tensorflow/python/keras/BUILD +++ b/tensorflow/python/keras/BUILD @@ -8,6 +8,7 @@ exports_files(["LICENSE"]) package(default_visibility = ["//visibility:public"]) load("//tensorflow:tensorflow.bzl", "py_test") +load("//tensorflow:tensorflow.bzl", "cuda_py_test") config_setting( name = "empty_condition", @@ -656,16 +657,17 @@ py_test( ], ) -py_test( +cuda_py_test( name = "multi_gpu_utils_test", - size = "medium", srcs = ["_impl/keras/utils/multi_gpu_utils_test.py"], - srcs_version = "PY2AND3", - tags = ["multi_gpu"], - deps = [ + additional_deps = [ ":keras", - "//tensorflow/python:client_testlib", "//third_party/py/numpy", + "//tensorflow/python:client_testlib", + ], + tags = [ + "guitar", + "multi_gpu", ], ) -- 2.7.4