Searched refs:list_physical_devices (Results 1 – 25 of 25) sorted by relevance
356 cpus = config.list_physical_devices('CPU')359 gpus = config.list_physical_devices('GPU')364 cpus = config.list_physical_devices('CPU')411 gpus = config.list_physical_devices('GPU')414 cpus = config.list_physical_devices('CPU')449 gpus = config.list_physical_devices('GPU')468 gpus = config.list_physical_devices('GPU')513 gpus = config.list_physical_devices('GPU')538 gpus = config.list_physical_devices('GPU')618 gpus = config.list_physical_devices('GPU')[all …]
307 def list_physical_devices(device_type=None): function335 return context.context().list_physical_devices(device_type)
40 cpu_device = config.list_physical_devices("CPU")[0]45 cpu_device = config.list_physical_devices("CPU")[0]
48 if config.list_physical_devices('GPU'):70 gpus = config.list_physical_devices('GPU')
204 physical_devices = config.list_physical_devices("CPU")
53 cpus = config.list_physical_devices("CPU")
1244 physical_gpus = context.context().list_physical_devices(device_type="GPU")
1367 config.list_physical_devices("CPU")[0],
122 gpus = config.list_physical_devices('GPU')136 gpus = config.list_physical_devices('GPU')
48 if config.list_physical_devices('GPU'):
951 for dev in self.list_physical_devices("GPU"):1128 def list_physical_devices(self, device_type=None): member in Context
56 cpus = config.list_physical_devices('CPU')
70 cpus = config.list_physical_devices('CPU')
134 cpus = config.list_physical_devices('CPU')
27 if tf.config.list_physical_devices('GPU'):
44 name: "list_physical_devices"
60 name: "list_physical_devices"
60 cpus = config_module.list_physical_devices('CPU')
46 gpu_devices = config.list_physical_devices('GPU')
63 physical_devices = config.list_physical_devices("CPU")567 if not config.list_physical_devices("XLA_CPU"):
57 gpus = config.list_physical_devices('GPU')
383 cpus = config.list_physical_devices('CPU')
53 …longer experimental: `tf.config.list_logical_devices`, `tf.config.list_physical_devices`, `tf.conf…