Searched defs:ThenMemcpyDeviceToDevice (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ | ||
| D | local_device_state.cc | 89 Status LocalDeviceState::ThenMemcpyDeviceToDevice( in ThenMemcpyDeviceToDevice() function in xla::LocalDeviceState |
| D | tpu_client.cc | 66 Status TpuDeviceState::ThenMemcpyDeviceToDevice( in ThenMemcpyDeviceToDevice() function in xla::__anon240436b40111::TpuDeviceState |