Searched refs:_set_device (Results 1 – 7 of 7) sorted by relevance
103 def _set_device(self, device): member in _FakeOperation
245 new_op._set_device('') # pylint: disable=protected-access287 op._set_device(coloc_device) # pylint: disable=protected-access
2230 def _set_device(self, device): # pylint: disable=redefined-outer-name member in Operation3648 op._set_device(colocation_op.device)
732 op._set_device("/job:goo/device:GPU:0")736 op._set_device(
857 def _set_device(self, device_name, device_spec): member in Context1799 ctx._set_device(new_device_name, new_device_spec) # pylint: disable=protected-access1809 ctx._set_device(old_device_name, old_device_spec) # pylint: disable=protected-access
495 def _set_device(self, device): member in TPUReplicateContext._EnterOutsideCompilationScope.FakeOp
849 def _set_device(self, device): member in _TfDeviceCaptureOp