Searched defs:on_done_with_host_buffer (Results 1 – 3 of 3) sorted by relevance
176 std::function<void()> on_done_with_host_buffer; in HandleNumpyArray() local
434 std::function<void()> on_done_with_host_buffer, PjRtDevice* device) { in BufferFromHostBuffer()
684 std::function<void()> on_done_with_host_buffer, PjRtDevice* device) { in BufferFromHostBuffer()