nnicolosi/Raptor
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

cose
Validate Operations / validate-operations (push) Has been cancelled
Details

Browse Source
This commit is contained in:
NiccoloN
2026-06-25 18:57:12 +02:00
parent be0bcc9dcc
commit 568fd90542
6 changed files with 647 additions and 179 deletions
Download Patch File Download Diff File Expand all files Collapse all files
validation/gen_network_runner.py
+9 -9
View File
@@ -6,15 +6,15 @@ from onnx import TensorProto
# ONNX dtype -> (ctype, printf, ONNX_TYPE_*)
DTYPES = {
TensorProto.FLOAT: ("float", "%g", "ONNX_TYPE_FLOAT"),
TensorProto.DOUBLE: ("double", "%g", "ONNX_TYPE_DOUBLE"),
TensorProto.INT64: ("int64_t", "%lld","ONNX_TYPE_INT64"),
TensorProto.INT32: ("int32_t", "%d", "ONNX_TYPE_INT32"),
TensorProto.UINT8: ("uint8_t", "%u", "ONNX_TYPE_UINT8"),
TensorProto.INT8: ("int8_t", "%d", "ONNX_TYPE_INT8"),
TensorProto.BOOL: ("uint8_t", "%u", "ONNX_TYPE_BOOL"), # stored as byte
TensorProto.FLOAT16: ("uint16_t", "%u", "ONNX_TYPE_FLOAT16"), # raw 16-bit
TensorProto.BFLOAT16:("uint16_t", "%u", "ONNX_TYPE_BFLOAT16"),
TensorProto.FLOAT: ("float", "%.9g", "ONNX_TYPE_FLOAT"),
TensorProto.DOUBLE: ("double", "%.17g", "ONNX_TYPE_DOUBLE"),
TensorProto.INT64: ("int64_t", "%lld", "ONNX_TYPE_INT64"),
TensorProto.INT32: ("int32_t", "%d", "ONNX_TYPE_INT32"),
TensorProto.UINT8: ("uint8_t", "%u", "ONNX_TYPE_UINT8"),
TensorProto.INT8: ("int8_t", "%d", "ONNX_TYPE_INT8"),
TensorProto.BOOL: ("uint8_t", "%u", "ONNX_TYPE_BOOL"),
TensorProto.FLOAT16: ("uint16_t", "%u", "ONNX_TYPE_FLOAT16"),
TensorProto.BFLOAT16:("uint16_t", "%u", "ONNX_TYPE_BFLOAT16"),
}
def esc(s): return s.replace("\\","\\\\").replace('"','\\"')