Searched defs:_set_device (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ | ||
| D | device_util.py | 83 def _set_device(self, device): member in _FakeOperation |
| /external/tensorflow/tensorflow/python/tpu/ | ||
| D | tpu.py | 288 def _set_device(self, device): member in TPUReplicateContext._EnterOutsideCompilationScope.FakeOp |
| /external/tensorflow/tensorflow/python/keras/ | ||
| D | backend.py | 532 def _set_device(self, device): member in _TfDeviceCaptureOp |
| /external/tensorflow/tensorflow/python/framework/ | ||
| D | ops.py | 2212 def _set_device(self, device): # pylint: disable=redefined-outer-name member in Operation |