diff options
-rw-r--r-- | src/TosaDeserialize.cpp | 6 | ||||
m--------- | third_party/serialization_lib | 0 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp index 82c107e..bd9fc9d 100644 --- a/src/TosaDeserialize.cpp +++ b/src/TosaDeserialize.cpp @@ -238,10 +238,10 @@ ConstructConstAttr(const mlir::RankedTensorType &output_type, break; } case DType_FP16: { - std::vector<float> float_data; - TosaSerializationHandler::ConvertU8toF16(data, out_size, float_data); + std::vector<half_float::half> half_data; + TosaSerializationHandler::ConvertU8toF16(data, out_size, half_data); value_attr = - mlir::DenseElementsAttr::get(output_type, llvm::ArrayRef(float_data)); + mlir::DenseElementsAttr::get(output_type, llvm::ArrayRef(half_data)); break; } case DType_UINT8: diff --git a/third_party/serialization_lib b/third_party/serialization_lib -Subproject 81db8ee8f580d30ec0ca53067df32ef046e6f09 +Subproject 758e73e117c5cef17f8f0b1c543efc1df953b2f |