View source on GitHub |
Device chooser for variables.
tf.contrib.framework.VariableDeviceChooser(
num_tasks=0, job_name='ps', device_type='CPU', device_index=0, replica=None
)
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.
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
__call__
__call__(
op
)
Call self as a function.