Python – tensorflow.DeviceSpec
Last Updated :
28 Jul, 2020
TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning neural networks.
DeviceSpec represents the specification of TensorFlow device. This specification might be partial. If a DeviceSpec is partially specified, it will be merged with other DeviceSpec`s according to the scope in which it is defined.
Syntax: tensorflow.DeviceSpec( job, replica, task, device_type, device_index )
Parameters:
- job(optional): It is a string which specifies the job name.
- replica(optional): It is an integer which specifies the replica index.
- task(optional): It is an integer which specifies the task index.
- device_type(optional): It can either be CPU or GPU.
- device_index(optional): It is an integer which specifies the device index.
Returns: It returns a DeviceSpec object.
Example 1:
Python3
import tensorflow as tf
device_spec = tf.DeviceSpec(job ="gfg", device_type ="GPU", device_index = 0)
print('DeviceSpec: ', device_spec)
|
Output:
DeviceSpec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5c1818ac8>
Example 2:
Python3
import tensorflow as tf
device_spec = tf.DeviceSpec(job ="gfg", device_type ="CPU", device_index = 0)
print('DeviceSpec: ', device_spec)
|
Output:
DeviceSpec: <tensorflow.python.framework.device_spec.DeviceSpecV2 object at 0x7fe5bb29d888>
Like Article
Suggest improvement
Share your thoughts in the comments
Please Login to comment...