We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bc1dd9a commit aefc71fCopy full SHA for aefc71f
tf2onnx/utils.py
@@ -38,15 +38,14 @@
38
onnx_pb.TensorProto.FLOAT: np.float32,
39
onnx_pb.TensorProto.FLOAT16: np.float16,
40
onnx_pb.TensorProto.DOUBLE: np.float64,
41
+ onnx_pb.TensorProto.INT64: np.int64,
42
onnx_pb.TensorProto.INT32: np.int32,
43
onnx_pb.TensorProto.INT16: np.int16,
44
onnx_pb.TensorProto.INT8: np.int8,
45
onnx_pb.TensorProto.UINT8: np.uint8,
46
onnx_pb.TensorProto.UINT16: np.uint16,
47
onnx_pb.TensorProto.UINT32: np.uint32,
48
onnx_pb.TensorProto.UINT64: np.uint64,
- onnx_pb.TensorProto.INT64: np.int64,
49
50
onnx_pb.TensorProto.BOOL: bool,
51
onnx_pb.TensorProto.COMPLEX64: np.complex64,
52
onnx_pb.TensorProto.COMPLEX128: np.complex128,
0 commit comments