Searched refs:on_delete_callback (Results 1 – 6 of 6) sorted by relevance
354 std::function<void()> on_delete_callback; in DLPackManagedTensorToBuffer() local356 on_delete_callback = [dlmt]() { dlmt->deleter(dlmt); }; in DLPackManagedTensorToBuffer()362 shape, device, on_delete_callback)); in DLPackManagedTensorToBuffer()
167 std::function<void()> on_delete_callback) in TrackedDeviceBuffer() argument174 on_delete_callback_(std::move(on_delete_callback)) {} in TrackedDeviceBuffer()
211 std::function<void()> on_delete_callback);
246 std::function<void()> on_delete_callback) = 0;
686 std::function<void()> on_delete_callback; in BufferFromHostBuffer() local693 on_delete_callback = std::move(on_done_with_host_buffer); in BufferFromHostBuffer()703 on_delete_callback = [staging_buffer, host_memory_allocator = in BufferFromHostBuffer()713 definition_events, std::move(on_delete_callback)); in BufferFromHostBuffer()960 std::function<void()> on_delete_callback) { in CreateViewOfDeviceBuffer() argument966 std::move(on_delete_callback)); in CreateViewOfDeviceBuffer()
202 std::function<void()> on_delete_callback) override;