Searched defs:ThenMemcpyDeviceToDevice (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ | ||
| H A D | local_device_state.cc | 100 Status LocalDeviceState::ThenMemcpyDeviceToDevice( in ThenMemcpyDeviceToDevice() function in xla::LocalDeviceState |
| H A D | tpu_client.cc | 72 Status TpuDeviceState::ThenMemcpyDeviceToDevice( in ThenMemcpyDeviceToDevice() function in xla::__anoncf75f5430111::TpuDeviceState |