View source on GitHub
|
Return all the available devices based on the device type.
tf.keras.distribution.list_devices(
device_type=None
)
Args | |
|---|---|
device_type
|
string, one of "cpu", "gpu" or "tpu".
Defaults to "gpu" or "tpu" if available when
device_type is not provided. Otherwise
will return the "cpu" devices.
|
Return | |
|---|---|
| List of devices that are available for distribute computation. |