Searched refs:kGPU0 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util_test.cc | 121 const char* kGPU0 = "/job:localhost/replica:0/task:0/device:GPU:0"; variable
129 CheckPickDeviceResult(kGPU0, false, {kGPU0, kGPU0}); in TEST()
133 CheckPickDeviceResult(kGPU0, false, {kGPU0, kCPU0}); in TEST()
142 CheckPickDeviceHasError(false, {kGPU0, kXPU1}); in TEST()
151 CheckPickDeviceHasError(false, {kGPU0, kGPU1}); in TEST()
153 CheckPickDeviceHasError(false, {kCPU0, kCPU1, kGPU0}); in TEST()
|
| D | mark_for_compilation_pass_test.cc | 1191 const char* kGPU0 = "/job:worker/replica:0/task:0/device:GPU:0"; variable
1207 FindNodeByName(graph.get(), "test/x")->set_assigned_device_name(kGPU0); in TEST()
1209 FindNodeByName(graph.get(), "test/z")->set_assigned_device_name(kGPU0); in TEST()
1232 FindNodeByName(graph.get(), "test/x")->set_assigned_device_name(kGPU0); in TEST()
1280 ->set_assigned_device_name(kGPU0); in TEST()
1281 FindNodeByName(graph.get(), var_handle_name)->set_assigned_device_name(kGPU0); in TEST()
1304 FindNodeByName(graph.get(), "test/b")->set_assigned_device_name(kGPU0); in TEST()
|