Skip to content

Commit 90c2b8b

Browse files
Expose ONNX type, size, and name constants in OMTensor (llvm#1175)
Signed-off-by: Gong Su <[email protected]> Co-authored-by: Alexandre Eichenberger <[email protected]>
1 parent 52fb808 commit 90c2b8b

File tree

1 file changed

+22
-22
lines changed

1 file changed

+22
-22
lines changed

src/Runtime/jni/src/com/ibm/onnxmlir/OMTensor.java

+22-22
Original file line numberDiff line numberDiff line change
@@ -12,33 +12,31 @@
1212

1313
public class OMTensor {
1414

15-
private final ByteOrder nativeEndian = ByteOrder.nativeOrder();
16-
1715
/* We can use enum but that creates another class
1816
* which complicates things for JNI.
1917
*
2018
* Values are standard ONNX data types defined in
2119
* https://github.com/onnx/onnx/blob/main/onnx/onnx.proto#L484
2220
*/
23-
final static int ONNX_TYPE_UNDEFINED = 0;
24-
final static int ONNX_TYPE_FLOAT = 1;
25-
final static int ONNX_TYPE_UINT8 = 2;
26-
final static int ONNX_TYPE_INT8 = 3;
27-
final static int ONNX_TYPE_UINT16 = 4;
28-
final static int ONNX_TYPE_INT16 = 5;
29-
final static int ONNX_TYPE_INT32 = 6;
30-
final static int ONNX_TYPE_INT64 = 7;
31-
final static int ONNX_TYPE_STRING = 8;
32-
final static int ONNX_TYPE_BOOL = 9;
33-
final static int ONNX_TYPE_FLOAT16 = 10;
34-
final static int ONNX_TYPE_DOUBLE = 11;
35-
final static int ONNX_TYPE_UINT32 = 12;
36-
final static int ONNX_TYPE_UINT64 = 13;
37-
final static int ONNX_TYPE_COMPLEX64 = 14;
38-
final static int ONNX_TYPE_COMPLEX128 = 15;
39-
final static int ONNX_TYPE_BFLOAT16 = 16;
40-
41-
final static int[] ONNX_TYPE_SIZE = new int[] {
21+
public final static int ONNX_TYPE_UNDEFINED = 0;
22+
public final static int ONNX_TYPE_FLOAT = 1;
23+
public final static int ONNX_TYPE_UINT8 = 2;
24+
public final static int ONNX_TYPE_INT8 = 3;
25+
public final static int ONNX_TYPE_UINT16 = 4;
26+
public final static int ONNX_TYPE_INT16 = 5;
27+
public final static int ONNX_TYPE_INT32 = 6;
28+
public final static int ONNX_TYPE_INT64 = 7;
29+
public final static int ONNX_TYPE_STRING = 8;
30+
public final static int ONNX_TYPE_BOOL = 9;
31+
public final static int ONNX_TYPE_FLOAT16 = 10;
32+
public final static int ONNX_TYPE_DOUBLE = 11;
33+
public final static int ONNX_TYPE_UINT32 = 12;
34+
public final static int ONNX_TYPE_UINT64 = 13;
35+
public final static int ONNX_TYPE_COMPLEX64 = 14;
36+
public final static int ONNX_TYPE_COMPLEX128 = 15;
37+
public final static int ONNX_TYPE_BFLOAT16 = 16;
38+
39+
public final static int[] ONNX_TYPE_SIZE = new int[] {
4240
0, /* UNDEFINED */
4341
4, /* FLOAT */
4442
1, /* UINT8 */
@@ -58,7 +56,7 @@ public class OMTensor {
5856
2, /* BFLOAT16 */
5957
};
6058

61-
final static String[] ONNX_TYPE_NAME = new String[] {
59+
public final static String[] ONNX_TYPE_NAME = new String[] {
6260
"UNDEFINED",
6361
"FLOAT",
6462
"UINT8",
@@ -78,6 +76,8 @@ public class OMTensor {
7876
"BFLOAT16",
7977
};
8078

79+
private final ByteOrder nativeEndian = ByteOrder.nativeOrder();
80+
8181
private ByteBuffer _data;
8282
private long[] _shape;
8383
private long[] _strides;

0 commit comments

Comments
 (0)