Searched refs:on_done_with_host_buffer (Results 1 – 4 of 4) sorted by relevance
650 std::function<void()> on_done_with_host_buffer, PjRtDevice* device) { in BufferFromHostBuffer() argument693 on_delete_callback = std::move(on_done_with_host_buffer); in BufferFromHostBuffer()700 if (on_done_with_host_buffer) { in BufferFromHostBuffer()701 on_done_with_host_buffer(); in BufferFromHostBuffer()749 if (on_done_with_host_buffer) { in BufferFromHostBuffer()750 on_done_with_host_buffer(); in BufferFromHostBuffer()751 on_done_with_host_buffer = nullptr; in BufferFromHostBuffer()768 on_done_with_host_buffer{ in BufferFromHostBuffer()769 std::move(on_done_with_host_buffer)}, in BufferFromHostBuffer()811 on_done_with_host_buffer{std::move(on_done_with_host_buffer)}]() { in BufferFromHostBuffer()[all …]
228 std::function<void()> on_done_with_host_buffer, PjRtDevice* device) = 0;
190 std::function<void()> on_done_with_host_buffer,
126 std::function<void()> on_done_with_host_buffer; in PjRtBufferFromPyval() local131 on_done_with_host_buffer = in PjRtBufferFromPyval()142 std::move(on_done_with_host_buffer), device)); in PjRtBufferFromPyval()