Function TVMFFIEnvGetTensorAllocator

• Defined in File c_env_api.h

Function Documentation

DLPackTensorAllocator TVMFFIEnvGetTensorAllocator()

FFI function get the current DLPack allocator stored in context.

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

Returns:

The current DLPack allocator.