pjrt
ApiMemoryLayout
ApiMemoryLayout
#include <layout_utils.h>class ApiMemoryLayoutDefined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:15
Public Methods
| Return | Name | Description |
|---|---|---|
ApiMemoryLayout | Defaulted constructor. | |
void | InitializeDenseRowMajorStrided | |
void | InitializeDenseRowMajorTiled | |
void | Reset inline | |
bool | is_valid const inline | |
const PJRT_Buffer_MemoryLayout & | c_layout const inline |
ApiMemoryLayout
ApiMemoryLayout() = defaultDefined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:17
Defaulted constructor.
InitializeDenseRowMajorStrided
void InitializeDenseRowMajorStrided(size_t rank, const int64_t * dims, size_t unit_stride_bytes)Defined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:19
InitializeDenseRowMajorTiled
void InitializeDenseRowMajorTiled(int64_t rank)Defined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:21
Reset
inline
inline void Reset()Defined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:22
is_valid
const inline
inline bool is_valid() constDefined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:24
c_layout
const inline
inline const PJRT_Buffer_MemoryLayout & c_layout() constDefined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:25
Private Attributes
c_layout_
PJRT_Buffer_MemoryLayout c_layout_Defined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:28
storage1_
std::vector< int64_t > storage1_Defined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:31
valid_
bool valid_ = falseDefined in integrations/pjrt/src/iree_pjrt/common/layout_utils.h:33
