LoadedExecutableInstance
LoadedExecutableInstance
#include <api_impl.h>class LoadedExecutableInstanceDefined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:406
Public Methods
| Return | Name | Description |
|---|---|---|
LoadedExecutableInstance inline | ||
~LoadedExecutableInstance inline | ||
operator PJRT_LoadedExecutable * inline | ||
const std::vector< DeviceInstance * > & | addressable_devices inline | |
iree_status_t | LoadAll | |
iree_status_t | GetDefaultResidentExecutable | |
iree_status_t | GetArgResultCount | |
iree_status_t | BatchExecute |
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
| Return | Name | Description |
|---|---|---|
void | BindApi 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
| Return | Name | Description |
|---|---|---|
ClientInstance & | client_ | |
ExecutableImage * | image_ | |
std::vector< DeviceInstance * > | addressable_devices_ | |
std::vector< ResidentExecutable > | resident_executables_ |
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
