Fix bug in DispatchOnTensorType macro (#4808)

This commit is contained in:
Hariharan Seshadri 2020-08-17 01:16:01 -07:00 committed by GitHub
parent 5899c1197a
commit ea3b4e1f8d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -112,7 +112,7 @@ constexpr ONNX_NAMESPACE::TensorProto_DataType ToTensorProtoElementType<BFloat16
function<int8_t>(__VA_ARGS__); \
break; \
case ONNX_NAMESPACE::TensorProto_DataType_UINT8: \
function<uint32_t>(__VA_ARGS__); \
function<uint8_t>(__VA_ARGS__); \
break; \
case ONNX_NAMESPACE::TensorProto_DataType_INT16: \
function<int16_t>(__VA_ARGS__); \