#include <allocator.h>

Inheritance diagram for onnxruntime::CPUAllocator:

Public Member Functions
	CPUAllocator (const OrtMemoryInfo &memory_info)

	CPUAllocator ()

void *	Alloc (size_t size) override

void	Free (void *p) override

Public Member Functions inherited from onnxruntime::IAllocator
	IAllocator (const OrtMemoryInfo &info)

virtual	~IAllocator ()=default

virtual void *	Reserve (size_t size)

const OrtMemoryInfo &	Info () const

virtual void	GetStats (AllocatorStats *)

void *	AllocArray (size_t nmemb, size_t size)

template<size_t alignment>
void *	AllocArrayWithAlignment (size_t nmemb, size_t size)

Additional Inherited Members
Static Public Member Functions inherited from onnxruntime::IAllocator
static bool	CalcMemSizeForArray (size_t nmemb, size_t size, size_t *out) noexcept

static bool	CalcMemSizeForArrayWithAlignment (size_t nmemb, size_t size, size_t alignment, size_t *out) noexcept

template<size_t alignment>
static bool	CalcMemSizeForArrayWithAlignment (size_t nmemb, size_t size, size_t *out) noexcept

template<typename T >
static IAllocatorUniquePtr< T >	MakeUniquePtr (std::shared_ptr< IAllocator > allocator, size_t count_or_bytes, bool use_reserve=false, Stream *stream=nullptr, WaitNotificationFn wait_fn=nullptr)

template<typename T >
static IAllocatorUniquePtr< T >	MakeUniquePtrFromOrtAllocator (OrtAllocator *ort_allocator, size_t count_or_bytes)

Detailed Description

Definition at line 251 of file allocator.h.

Constructor & Destructor Documentation

onnxruntime::CPUAllocator::CPUAllocator ( const OrtMemoryInfo & memory_info )

inlineexplicit

Definition at line 253 of file allocator.h.

onnxruntime::CPUAllocator::CPUAllocator ( )

inline

Definition at line 255 of file allocator.h.

void* onnxruntime::CPUAllocator::Alloc ( size_t size )

overridevirtual

Allocate memory of the specified size. If size is 0, nullptr is returned. If allocation fails, an exception is thrown.

Remarks: Use SafeInt when calculating the size of memory to allocate using Alloc.

void onnxruntime::CPUAllocator::Free ( void * p )

overridevirtual

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