Home
last modified time | relevance | path

Searched defs:CopyToRemoteDeviceScattered (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.h453 Status CopyToRemoteDeviceScattered( in CopyToRemoteDeviceScattered() function
Dpjrt_stream_executor_client.h283 virtual Status CopyToRemoteDeviceScattered( in CopyToRemoteDeviceScattered() function
Dpjrt_stream_executor_client.cc1548 Status PjRtStreamExecutorBuffer::CopyToRemoteDeviceScattered( in CopyToRemoteDeviceScattered() function in xla::PjRtStreamExecutorBuffer