parameters: (self,
                 num_tasks=0,
                 job_name='ps',
                 device_type='CPU',
                 device_index=0)
docstring: 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.
docstring_summary: Initialize VariableDeviceChooser.
docstring_tokens: Initialize VariableDeviceChooser .
function: def __init__(self,
                 num_tasks=0,
                 job_name='ps',
                 device_type='CPU',
                 device_index=0):
      """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.
      """
      self._job_name = job_name
      self._device_type = device_type
      self._device_index = device_index
      self._num_tasks = num_tasks
      self._next_task_id = 0
function_tokens: def __init__ ( self , num_tasks = 0 , job_name = 'ps' , device_type = 'CPU' , device_index = 0 ) : self . _job_name = job_name self . _device_type = device_type self . _device_index = device_index self . _num_tasks = num_tasks self . _next_task_id = 0