Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.config.LogicalDevice

View source on GitHub

Abstraction for a logical device initialized by the runtime.

@staticmethod
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.

Fields:

  • 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".

Attributes:

  • name
  • device_type