View source on GitHub
|
Abstraction for a logical device initialized by the runtime.
tf.config.LogicalDevice(
_cls, name, device_type
)
A tf.config.LogicalDevice corresponds to an initialized logical device on a
tf.config.PhysicalDevice or a remote device visible to the cluster. Tensors
and operations can be placed on a specific logical device by calling
tf.device with a specified tf.config.LogicalDevice.
name: The fully qualified name of the device. Can be used for Op or function
placement.device_type: String declaring the type of device such as "CPU" or "GPU".namedevice_type