Searched refs:list_logical_devices (Results 1 – 25 of 31) sorted by relevance
12
64 [device.name for device in config.list_logical_devices()])80 [device.name for device in config.list_logical_devices()])
63 devices = config.list_logical_devices(device_type=device_type)
355 with ctx.device(ctx.list_logical_devices('CPU')[0]):360 gpus = ctx.list_logical_devices('GPU')375 cpus = ctx.list_logical_devices('CPU')
396 vcpus = config.list_logical_devices('CPU')448 self.assertEqual(len(config.list_logical_devices('XLA_GPU')), 0)513 logical_devices = config.list_logical_devices()538 logical_gpus = config.list_logical_devices('GPU')584 logical_gpus = config.list_logical_devices('GPU')702 gpus = config.list_logical_devices('GPU')707 gpus = config.list_logical_devices('GPU')725 gpus = config.list_logical_devices('GPU')
422 def list_logical_devices(device_type=None): function452 return context.context().list_logical_devices(device_type=device_type)
30 gpus = context.context().list_logical_devices("GPU")
63 config.list_logical_devices("CPU")[0].name)
176 devices = config.list_logical_devices("GPU")179 return devices or config.list_logical_devices("CPU")
1393 logical_gpus = context.context().list_logical_devices(device_type="GPU")1400 machine_devices = context.context().list_logical_devices()
60 name: "list_logical_devices"
64 name: "list_logical_devices"
140 num_gpus = len(config.list_logical_devices('GPU'))
80 logical_devices = config.list_logical_devices()
140 if context.executing_eagerly() and tf_config.list_logical_devices('GPU'):
410 if config.list_logical_devices('GPU'):527 (device_type is None and config.list_logical_devices('GPU'))) and1136 if config.list_logical_devices('GPU'):1255 (device_type is None and config.list_logical_devices('GPU'))) and
434 if context.executing_eagerly() and tf_config.list_logical_devices("GPU"):543 if context.executing_eagerly() and tf_config.list_logical_devices("GPU"):719 if context.executing_eagerly() and tf_config.list_logical_devices("GPU"):925 if context.executing_eagerly() and tf_config.list_logical_devices("GPU"):
368 d for d in framework_config.list_logical_devices()
45 logical_devices = config.list_logical_devices()
55 devices = config.list_logical_devices(device_type)60 config.list_logical_devices())
56 if config.list_logical_devices('TPU'):
101 " devices = tf.config.list_logical_devices('GPU')\n",107 " devices = tf.config.list_logical_devices('CPU')\n",
231 "gpus = tf.config.list_logical_devices(\"GPU\")\n",
355 device_list = config.list_logical_devices("GPU")