Home
last modified time | relevance | path

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

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