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.
[null,null,["Last updated 2024-06-07 UTC."],[],[],null,["# tf.keras.distribution.list_devices\n\n\u003cbr /\u003e\n\n|-----------------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/keras-team/keras/tree/v3.3.3/keras/src/distribution/distribution_lib.py#L26-L41) |\n\nReturn all the available devices based on the device type. \n\n tf.keras.distribution.list_devices(\n device_type=None\n )\n\n| **Note:** in a distributed setting, global devices are returned.\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|---------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| `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. |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Return ------ ||\n|---|---|\n| List of devices that are available for distribute computation. ||\n\n\u003cbr /\u003e"]]
View source on GitHub