diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/TosaDeserialize.cpp | 6 |
1 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: |