Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
View the Project on GitHub onnx/onnx-mlir
This project is maintained by onnx
Hosted on GitHub Pages — Theme by orderedlist
onnx-mlir
|
Public Member Functions | |
OMTensor (byte[] data, long[] shape, boolean flag) | |
OMTensor (byte[] data, long[] shape) | |
OMTensor (short[] data, long[] shape) | |
OMTensor (int[] data, long[] shape) | |
OMTensor (long[] data, long[] shape) | |
OMTensor (float[] data, long[] shape) | |
OMTensor (double[] data, long[] shape) | |
byte[] | getBoolData () |
void | setBoolData (byte[] data) |
byte[] | getByteData () |
void | setByteData (byte[] data) |
short[] | getShortData () |
void | setShortData (short[] data) |
int[] | getIntData () |
void | setIntData (int[] data) |
long[] | getLongData () |
void | setLongData (long[] data) |
float[] | getFloatData () |
void | setFloatData (float[] data) |
double[] | getDoubleData () |
void | setDoubleData (double[] data) |
long[] | getShape () |
void | setShape (long[] shape) |
long[] | getStrides () |
void | setStrides (long[] strides) |
int | getDataType () |
void | setDataType (int dataType) |
long | getBufferSize () |
long | getRank () |
long | getNumElems () |
Static Public Attributes | |
static final int | ONNX_TYPE_UNDEFINED = 0 |
static final int | ONNX_TYPE_FLOAT = 1 |
static final int | ONNX_TYPE_UINT8 = 2 |
static final int | ONNX_TYPE_INT8 = 3 |
static final int | ONNX_TYPE_UINT16 = 4 |
static final int | ONNX_TYPE_INT16 = 5 |
static final int | ONNX_TYPE_INT32 = 6 |
static final int | ONNX_TYPE_INT64 = 7 |
static final int | ONNX_TYPE_STRING = 8 |
static final int | ONNX_TYPE_BOOL = 9 |
static final int | ONNX_TYPE_FLOAT16 = 10 |
static final int | ONNX_TYPE_DOUBLE = 11 |
static final int | ONNX_TYPE_UINT32 = 12 |
static final int | ONNX_TYPE_UINT64 = 13 |
static final int | ONNX_TYPE_COMPLEX64 = 14 |
static final int | ONNX_TYPE_COMPLEX128 = 15 |
static final int | ONNX_TYPE_BFLOAT16 = 16 |
static final int | ONNX_TYPE_FLOAT8E4M3FN = 17 |
static final int | ONNX_TYPE_FLOAT8E4M3FNUZ = 18 |
static final int | ONNX_TYPE_FLOAT8E5M2 = 19 |
static final int | ONNX_TYPE_FLOAT8E5M2FNUZ = 20 |
static final int | LAST_ONNX_TYPE = 20 |
static final int[] | ONNX_TYPE_SIZE |
static final String[] | ONNX_TYPE_NAME |
Protected Member Functions | |
OMTensor (ByteBuffer data, long[] shape, ByteOrder endian, int dataType) | |
OMTensor (ByteBuffer data, long[] shape, long[] strides, int dataType) | |
ByteBuffer | getData () |
void | setData (ByteBuffer data) |
Class describing the runtime information such as rank, shape, strides, data type, etc. associated with a tensor input/output.
|
inline |
Constructor
data | byte data array for tensor |
shape | data shape array |
flag | true for boolean tensor, false for byte tensor |
|
inline |
Constructor
data | byte data array for tensor |
shape | data shape array |
|
inline |
Constructor
data | short data array for tensor |
shape | data shape array |
|
inline |
Constructor
data | int data array for tensor |
shape | data shape array |
|
inline |
Constructor
data | long data array for tensor |
shape | data shape array |
|
inline |
Constructor
data | float data array for tensor |
shape | data shape array |
|
inline |
Constructor
data | double data array for tensor |
shape | data shape array |
|
inlineprotected |
Constructor (For OMRunner only. Not intended for end user)
data | data buffer |
shape | data shape |
dataType | data type |
endian | data endian |
|
inlineprotected |
Constructor (For JNI wrapper only. Not intended for end user)
data | data buffer |
shape | data shape |
strides | data stride |
dataType | data type |
|
inline |
Bool data getter
|
inline |
Data buffer size getter
|
inline |
Byte data getter
|
inlineprotected |
Raw data getter (For JNI wrapper only. Not intended for end user)
|
inline |
Data type getter
|
inline |
Double data getter
|
inline |
Float data getter
|
inline |
Int data getter
|
inline |
Long data getter
|
inline |
Number of elements getter
|
inline |
Rank getter (return long to be consistent with C/C++ API)
|
inline |
Data shape getter
|
inline |
Short data getter
|
inline |
Data strides getter
|
inline |
Bool data setter
data | bool array to be set |
|
inline |
Byte data setter
data | byte array to be set |
|
inlineprotected |
Raw data setter (For JNI wrapper only. Not intended for end user)
data | raw data to be set |
|
inline |
Data type setter
dataType | data type to be set |
|
inline |
Double data setter
data | double array to be set |
|
inline |
Float data setter
data | float array to be set |
|
inline |
Int data setter
data | int array to be set |
|
inline |
Long data setter
data | long array to be set |
|
inline |
Data shape setter
shape | data shape array to be set |
|
inline |
Short data setter
data | short array to be set |
|
inline |
Data strides setter
strides | data strides array to be set |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |
|
static |