Searched refs:_set_device_from_string (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | device_util.py | 133 def _set_device_from_string(self, device_str): member in _FakeOperation |
| /external/tensorflow/tensorflow/python/framework/ | ||
| D | ops.py | 2370 self._set_device_from_string(compat.as_str(_device_string(device))) |
| /external/tensorflow/tensorflow/python/tpu/ | ||
| D | tpu.py | 525 def _set_device_from_string(self, device_str): member in TPUReplicateContext._EnterOutsideCompilationScope.FakeOp |
| /external/tensorflow/tensorflow/python/keras/ | ||
| D | backend.py | 852 def _set_device_from_string(self, device_str): member in _TfDeviceCaptureOp |