Home
last modified time | relevance | path

Searched defs:_set_device (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddevice_util.py130 def _set_device(self, device): member in _FakeOperation
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu.py519 def _set_device(self, device): member in TPUReplicateContext._EnterOutsideCompilationScope.FakeOp
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py982 def _set_device(self, device_name, device_spec): member in Context
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dbackend.py846 def _set_device(self, device): member in _TfDeviceCaptureOp
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dops.py2364 def _set_device(self, device): # pylint: disable=redefined-outer-name member in Operation