pjrt

MemoryInstance

MemoryInstance

#include <api_impl.h>
class MemoryInstance

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:200

Public Methods

ReturnNameDescription
MemoryInstance inline
intId
std::string_viewKind
std::string_viewDebugString
std::string_viewToString
DeviceInstance *device inline

MemoryInstance

inline

inline MemoryInstance(DeviceInstance * device)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:202


Id

int Id()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:210


Kind

std::string_view Kind()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:211


DebugString

std::string_view DebugString()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:212


ToString

std::string_view ToString()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:213


device

inline

inline DeviceInstance * device()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:215

Public Static Methods

ReturnNameDescription
voidBindApi static
MemoryInstance *Unwrap static inline

BindApi

static

static void BindApi(PJRT_Api * api)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:204


Unwrap

static inline

static inline MemoryInstance * Unwrap(PJRT_Memory * memory)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:206

Private Attributes

ReturnNameDescription
DeviceInstance *device_

device_

DeviceInstance * device_

Type: DeviceInstance *

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:218