Lines Matching refs:devices
57 self.devices = ['/cpu:0'] + gpu_devices
58 self._num_devices = len(self.devices)
84 logging.info('devices: %s', str(self.devices))
88 # Creates virtual GPU devices
129 devices,
135 for i in range(len(devices)):
136 with ops.device(devices[i]):
144 with ops.device(devices[op[2]]):
175 sess, random_seed, op_placement, self.devices, debug_mode=debug_mode)
189 logging.error('Devices: %s', self.devices)
206 # adding virtual devices in the future, thus must be called within a
207 # context of a session within which virtual devices are created. Same in
217 for d in self._util.devices:
229 devices = [d.device for d in step_stats.dev_stats]
230 self.assertTrue('/job:localhost/replica:0/task:0/device:CPU:0' in devices)
231 self.assertTrue('/job:localhost/replica:0/task:0/device:GPU:0' in devices)
232 self.assertTrue('/job:localhost/replica:0/task:0/device:GPU:1' in devices)
233 self.assertTrue('/job:localhost/replica:0/task:0/device:GPU:2' in devices)
