Searched refs:kGPU1 (Results 1 – 2 of 2) sorted by relevance
125 const char* kGPU1 = "/job:localhost/replica:0/task:0/device:GPU:1"; variable151 CheckPickDeviceHasError(false, {kGPU0, kGPU1}); in TEST()
1193 const char* kGPU1 = "/job:worker/replica:0/task:0/device:GPU:1"; variable1233 FindNodeByName(graph.get(), "test/y")->set_assigned_device_name(kGPU1); in TEST()