Set the device of this operation. Args: device: string or device.. The device to set.
(self, device)
| 1999 | return tf_input |
| 2000 | |
| 2001 | def _set_device(self, device): # pylint: disable=redefined-outer-name |
| 2002 | """Set the device of this operation. |
| 2003 | |
| 2004 | Args: |
| 2005 | device: string or device.. The device to set. |
| 2006 | """ |
| 2007 | self._set_device_from_string(compat.as_str(_device_string(device))) |
| 2008 | |
| 2009 | def _set_device_from_string(self, device_str): |
| 2010 | """Fast path to set device if the type is known to be a string. |