pjrt

LoadedExecutableInstance

LoadedExecutableInstance

#include <api_impl.h>
class LoadedExecutableInstance

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

Public Methods

ReturnNameDescription
LoadedExecutableInstance inline
~LoadedExecutableInstance inline
operator PJRT_LoadedExecutable * inline
const std::vector< DeviceInstance * > &addressable_devices inline
iree_status_tLoadAll
iree_status_tGetDefaultResidentExecutable
iree_status_tGetArgResultCount
iree_status_tBatchExecute

LoadedExecutableInstance

inline

inline LoadedExecutableInstance(ClientInstance & client, ExecutableImage * image, const std::vector< DeviceInstance * > & addressable_devices)

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


~LoadedExecutableInstance

inline

inline ~LoadedExecutableInstance()

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


operator PJRT_LoadedExecutable *

inline

inline operator PJRT_LoadedExecutable *()

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


addressable_devices

inline

inline const std::vector< DeviceInstance * > & addressable_devices()

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


LoadAll

iree_status_t LoadAll()

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


GetDefaultResidentExecutable

iree_status_t GetDefaultResidentExecutable(ResidentExecutable ** out_loaded)

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


GetArgResultCount

iree_status_t GetArgResultCount(iree_host_size_t * out_arg_count, iree_host_size_t * out_result_count)

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


BatchExecute

iree_status_t BatchExecute(PJRT_LoadedExecutable_Execute_Args * args)

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

Public Static Methods

ReturnNameDescription
voidBindApi static
LoadedExecutableInstance *Unwrap static inline

BindApi

static

static void BindApi(PJRT_Api * api)

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


Unwrap

static inline

static inline LoadedExecutableInstance * Unwrap(PJRT_LoadedExecutable * exe)

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

Private Attributes


client_

ClientInstance & client_

Type: ClientInstance &

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


image_

ExecutableImage * image_

Type: ExecutableImage *

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


addressable_devices_

std::vector< DeviceInstance * > addressable_devices_

Type: std::vector< DeviceInstance * >

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


resident_executables_

std::vector< ResidentExecutable > resident_executables_

Type: std::vector< ResidentExecutable >

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