Abstraction for a logical device initialized by the runtime.
tf.config.LogicalDevice( 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".Attributes | |
---|---|
name | |
device_type |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/config/LogicalDevice