Class VariableDeviceChooser
Defined in tensorflow/contrib/framework/python/ops/variables.py
.
Device chooser for variables.
When using a parameter server it will assign them in a round-robin fashion. When not using a parameter server it allows GPU or CPU placement.
__init__
__init__(
num_tasks=0,
job_name='ps',
device_type='CPU',
device_index=0,
replica=None
)
Initialize VariableDeviceChooser.
Usage: To use with 2 parameter servers: VariableDeviceChooser(2)
To use without parameter servers: VariableDeviceChooser() VariableDeviceChooser(device_type='GPU') # For GPU placement
Args:
num_tasks
: number of tasks.job_name
: String, a name for the parameter server job.device_type
: Optional device type string (e.g. "CPU" or "GPU")device_index
: int. Optional device index. If left unspecified, device represents 'any' device_index.
Methods
tf.contrib.framework.VariableDeviceChooser.__call__
__call__(op)
Call self as a function.