Home | History | Annotate | Download | only in costs

Lines Matching refs:gpu_device

32   DeviceProperties gpu_device;
33   gpu_device.set_type("GPU");
34   devices["/job:localhost/replica:0/task:0/device:GPU:0"] = gpu_device;
62   DeviceProperties gpu_device;
63   gpu_device.set_type("GPU");
64   devices["/GPU:0"] = gpu_device;
121     DeviceProperties gpu_device;
122     gpu_device.set_type("GPU");
124                             "/replica:0/task:0/device:GPU:0")] = gpu_device;
178   DeviceProperties gpu_device;
179   gpu_device.set_type("GPU");
184         gpu_device;
195   DeviceProperties gpu_device;
196   gpu_device.set_type("GPU");
202           gpu_device;
290   DeviceProperties gpu_device;
291   gpu_device.set_type("GPU");
292   devices["/job:my_job/replica:0/task:0/device:GPU:0"] = gpu_device;