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