12
12
13
13
public class OMTensor {
14
14
15
- private final ByteOrder nativeEndian = ByteOrder .nativeOrder ();
16
-
17
15
/* We can use enum but that creates another class
18
16
* which complicates things for JNI.
19
17
*
20
18
* Values are standard ONNX data types defined in
21
19
* https://github.com/onnx/onnx/blob/main/onnx/onnx.proto#L484
22
20
*/
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 [] {
42
40
0 , /* UNDEFINED */
43
41
4 , /* FLOAT */
44
42
1 , /* UINT8 */
@@ -58,7 +56,7 @@ public class OMTensor {
58
56
2 , /* BFLOAT16 */
59
57
};
60
58
61
- final static String [] ONNX_TYPE_NAME = new String [] {
59
+ public final static String [] ONNX_TYPE_NAME = new String [] {
62
60
"UNDEFINED" ,
63
61
"FLOAT" ,
64
62
"UINT8" ,
@@ -78,6 +76,8 @@ public class OMTensor {
78
76
"BFLOAT16" ,
79
77
};
80
78
79
+ private final ByteOrder nativeEndian = ByteOrder .nativeOrder ();
80
+
81
81
private ByteBuffer _data ;
82
82
private long [] _shape ;
83
83
private long [] _strides ;
0 commit comments