tvm
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Static Public Member Functions | List of all members
tvm::runtime::vm::MemoryManager Class Reference

#include <memory_manager.h>

Collaboration diagram for tvm::runtime::vm::MemoryManager:

Static Public Member Functions

static MemoryManagerGlobal ()
 
static AllocatorGetOrCreateAllocator (Device dev, AllocatorType type)
 Get or create an allocator given the context and allocator type. More...
 
static AllocatorGetAllocator (Device dev)
 Get an allocator given the context. More...
 

Member Function Documentation

◆ GetAllocator()

static Allocator* tvm::runtime::vm::MemoryManager::GetAllocator ( Device  dev)
static

Get an allocator given the context.

Parameters
devThe TVM device
Returns
The memory allocator.

◆ GetOrCreateAllocator()

static Allocator* tvm::runtime::vm::MemoryManager::GetOrCreateAllocator ( Device  dev,
AllocatorType  type 
)
static

Get or create an allocator given the context and allocator type.

Parameters
devThe TVM device
typeThe allocator type
Returns
The memory allocator.

◆ Global()

static MemoryManager* tvm::runtime::vm::MemoryManager::Global ( )
static

The documentation for this class was generated from the following file: