Searched refs:on_done_with_host_buffer (Results 1 – 7 of 7) sorted by relevance
495 std::function<void()> on_done_with_host_buffer, PjRtDevice* device) { in BufferFromHostBuffer() argument519 on_delete_callback = std::move(on_done_with_host_buffer); in BufferFromHostBuffer()542 if (on_done_with_host_buffer) { in BufferFromHostBuffer()543 on_done_with_host_buffer(); in BufferFromHostBuffer()544 on_done_with_host_buffer = nullptr; in BufferFromHostBuffer()553 if (on_done_with_host_buffer) { in BufferFromHostBuffer()554 on_done_with_host_buffer(); in BufferFromHostBuffer()555 on_done_with_host_buffer = nullptr; in BufferFromHostBuffer()565 on_done_with_host_buffer = in BufferFromHostBuffer()566 std::move(on_done_with_host_buffer)]() mutable { in BufferFromHostBuffer()[all …]
694 std::function<void()> on_done_with_host_buffer, PjRtDevice* device) { in BufferFromHostBuffer() argument748 on_delete_callback = std::move(on_done_with_host_buffer); in BufferFromHostBuffer()756 if (on_done_with_host_buffer) { in BufferFromHostBuffer()757 on_done_with_host_buffer(); in BufferFromHostBuffer()822 if (on_done_with_host_buffer) { in BufferFromHostBuffer()823 on_done_with_host_buffer(); in BufferFromHostBuffer()824 on_done_with_host_buffer = nullptr; in BufferFromHostBuffer()840 on_done_with_host_buffer{std::move(on_done_with_host_buffer)}, in BufferFromHostBuffer()890 on_done_with_host_buffer{std::move(on_done_with_host_buffer)}]() { in BufferFromHostBuffer()891 if (on_done_with_host_buffer) { in BufferFromHostBuffer()[all …]
199 std::function<void()> on_done_with_host_buffer, in BufferFromHostBuffer() argument204 on_done_with_host_buffer, PjRtCApiDevice::GetWrapped(device))); in BufferFromHostBuffer()
187 std::function<void()> on_done_with_host_buffer,
688 std::function<void()> on_done_with_host_buffer, PjRtDevice* device) = 0;
237 std::function<void()> on_done_with_host_buffer,
186 std::function<void()> on_done_with_host_buffer; in HandleNumpyArray() local190 on_done_with_host_buffer = in HandleNumpyArray()202 std::move(on_done_with_host_buffer), to_device)); in HandleNumpyArray()