diff options
author | 2017-11-01 16:15:20 -0700 | |
---|---|---|
committer | 2017-11-01 16:20:28 -0700 | |
commit | 117bcd9cb5f3e55ce1fcc09a0bb4963c32bad8ce (patch) | |
tree | 3d9a1d16b4bb78057e1a613ab01858d0d0fbcf22 /tensorflow/core/common_runtime/function_test.cc | |
parent | 70698a168669e0335872ce9248a6c496328d7871 (diff) |
Adding support for local device names for ProcessFLR. Now one can specify a remote target as /device:CPU:0 or /device:GPU:0 etc.
PiperOrigin-RevId: 174252575
Diffstat (limited to 'tensorflow/core/common_runtime/function_test.cc')
-rw-r--r-- | tensorflow/core/common_runtime/function_test.cc | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/core/common_runtime/function_test.cc b/tensorflow/core/common_runtime/function_test.cc index b77a8f50c4..d183bf7c97 100644 --- a/tensorflow/core/common_runtime/function_test.cc +++ b/tensorflow/core/common_runtime/function_test.cc @@ -939,9 +939,8 @@ TEST_F(FunctionLibraryRuntimeTest, Gradient_AddSum) { TEST_F(FunctionLibraryRuntimeTest, CrossDevice) { Init({test::function::FindDevice()}); FunctionLibraryRuntime::Handle handle; - TF_CHECK_OK(Instantiate( - flr0_, "FindDevice", - {{"_target", "/job:localhost/replica:0/task:0/cpu:1"}}, &handle)); + TF_CHECK_OK(Instantiate(flr0_, "FindDevice", {{"_target", "/device:CPU:1"}}, + &handle)); Tensor y; FunctionLibraryRuntime::Options opts; |