Function TVMFFIEnvGetDLPackManagedTensorAllocator

• Defined in File c_env_api.h

Function Documentation

DLPackManagedTensorAllocator TVMFFIEnvGetDLPackManagedTensorAllocator()

FFI function get the current DLPackManagedTensorAllocator stored in context.

This function first queries the global context, and if not found, queries the thread-local context.

Returns:

The current setted DLPackManagedTensorAllocator