Home
last modified time | relevance | path

Searched defs:CopyToRemoteDevice (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_c_api_client.h367 void CopyToRemoteDevice(absl::string_view serialized_descriptor, in CopyToRemoteDevice() function
Dtfrt_cpu_pjrt_client.h322 void CopyToRemoteDevice(absl::string_view serialized_descriptor, in CopyToRemoteDevice() function
Dpjrt_stream_executor_client.h303 virtual void CopyToRemoteDevice( in CopyToRemoteDevice() function
Dpjrt_stream_executor_client.cc1555 void PjRtStreamExecutorBuffer::CopyToRemoteDevice( in CopyToRemoteDevice() function in xla::PjRtStreamExecutorBuffer
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_buffer.cc211 std::pair<Status, bool> PyBuffer::CopyToRemoteDevice( in CopyToRemoteDevice() function in xla::PyBuffer