Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.h199 std::function<void()> on_done_with_host_buffer, in BufferFromHostBuffer()
H A Dtfrt_cpu_pjrt_client.cc495 std::function<void()> on_done_with_host_buffer, PjRtDevice* device) { in BufferFromHostBuffer()
H A Dpjrt_stream_executor_client.cc694 std::function<void()> on_done_with_host_buffer, PjRtDevice* device) { in BufferFromHostBuffer()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_values.cc186 std::function<void()> on_done_with_host_buffer; in HandleNumpyArray() local