Class DLManagedMemoryBuffer

class DLManagedMemoryBuffer

Class to wrap the deleter of a DLManagedTensor.

This class is used with DLManagedTensorCtx class to wrap the DLManagedTensor.

A shared pointer to this class in DLManagedTensorCtx class is used as the deleter of the DLManagedTensorCtx::memory_ref

When the last reference to the DLManagedTensorCtx object is released, DLManagedTensorCtx::memory_ref will also be destroyed, which will call the deleter function of the DLManagedTensor object.

Public Functions

explicit DLManagedMemoryBuffer(DLManagedTensor *self)

~DLManagedMemoryBuffer()

© Copyright 2022-2023, NVIDIA. Last updated on Sep 13, 2023.