Home
last modified time | relevance | path

Searched refs:_set_device_from_string (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddevice_util.py133 def _set_device_from_string(self, device_str): member in _FakeOperation
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dops.py2370 self._set_device_from_string(compat.as_str(_device_string(device)))
2372 def _set_device_from_string(self, device_str): member in Operation
4801 op._set_device_from_string(device_string)
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu.py525 def _set_device_from_string(self, device_str): member in TPUReplicateContext._EnterOutsideCompilationScope.FakeOp
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dbackend.py852 def _set_device_from_string(self, device_str): member in _TfDeviceCaptureOp