Searched refs:_set_device (Results 1 – 6 of 6) sorted by relevance
224 new_op._set_device(device_function(new_op))
338 op._set_device(coloc_device) # pylint: disable=protected-access756 op._set_device(coloc_device) # pylint: disable=protected-access
1855 def _set_device(self, device): # pylint: disable=redefined-outer-name member in Operation3358 op._set_device(colocation_op.device) # pylint: disable=protected-access4186 op._set_device(device_function(op)) # pylint: disable=protected-access
264 op._set_device("/job:goo/device:GPU:0")268 op._set_device(
1855 c.op._set_device('/cpu:0') # pylint: disable=protected-access1923 c.op._set_device('/cpu:0') # pylint: disable=protected-access1944 c.op._set_device('/cpu:0') # pylint: disable=protected-access
435 def _set_device(self, device): member in _TfDeviceCaptureOp