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