Searched defs:_set_device (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | device_util.py | 103 def _set_device(self, device): member in _FakeOperation |
| /external/tensorflow/tensorflow/python/tpu/ | ||
| D | tpu.py | 495 def _set_device(self, device): member in TPUReplicateContext._EnterOutsideCompilationScope.FakeOp |
| /external/tensorflow/tensorflow/python/eager/ | ||
| D | context.py | 857 def _set_device(self, device_name, device_spec): member in Context |
| /external/tensorflow/tensorflow/python/keras/ | ||
| D | backend.py | 849 def _set_device(self, device): member in _TfDeviceCaptureOp |
| /external/tensorflow/tensorflow/python/framework/ | ||
| D | ops.py | 2230 def _set_device(self, device): # pylint: disable=redefined-outer-name member in Operation |