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