'Device to use to build the inputs.
Returns:
A value suitable for `tf.device()`.'

def inputs_device(self):

    device = ''
    if (self._num_ps_tasks > 0):
        device += self._worker_device
    device += '/device:CPU:0'
    return device
