Searched refs:_set_device (Results 1 – 7 of 7) sorted by relevance
92 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
2024 def _set_device(self, device): # pylint: disable=redefined-outer-name member in Operation3426 op._set_device(colocation_op.device)
572 op._set_device("/job:goo/device:GPU:0")576 op._set_device(
762 def _set_device(self, device_name, device_spec): member in Context1597 ctx._set_device(new_device_name, new_device_spec) # pylint: disable=protected-access1607 ctx._set_device(old_device_name, old_device_spec) # pylint: disable=protected-access
442 def _set_device(self, device): member in TPUReplicateContext._EnterOutsideCompilationScope.FakeOp
617 def _set_device(self, device): member in _TfDeviceCaptureOp