Home
last modified time | relevance | path

Searched defs:MakeCrossHostReceiveBuffers (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.h233 MakeCrossHostReceiveBuffers(absl::Span<const Shape> shapes, in MakeCrossHostReceiveBuffers() function
H A Dtfrt_cpu_pjrt_client.h194 MakeCrossHostReceiveBuffers(absl::Span<const Shape> shapes, in MakeCrossHostReceiveBuffers() function
H A Dpjrt_stream_executor_client.cc1000 PjRtStreamExecutorClient::MakeCrossHostReceiveBuffers( in MakeCrossHostReceiveBuffers() function in xla::PjRtStreamExecutorClient
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_client.cc272 PyClient::MakeCrossHostReceiveBuffers(absl::Span<const Shape> shapes, in MakeCrossHostReceiveBuffers() function in xla::PyClient