diff options
| -rw-r--r-- | reference_model/src/subgraph_traverser.cc | 6 | ||||
| m--------- | thirdparty/serialization_lib | 0 |
2 files changed, 3 insertions, 3 deletions
diff --git a/reference_model/src/subgraph_traverser.cc b/reference_model/src/subgraph_traverser.cc index 33a9b94..52b1806 100644 --- a/reference_model/src/subgraph_traverser.cc +++ b/reference_model/src/subgraph_traverser.cc @@ -564,8 +564,7 @@ int SubgraphTraverser::allocateTensor(std::string name) } break; case DType_FP16: { - // Interpret f16 data as float - std::vector<float> f16_data; + std::vector<half_float::half> f16_data; TosaSerializationHandler::ConvertU8toF16(ts->GetData(), tensor->getElementCount(), f16_data); if (tensor->getDtype() == TOSA_REF_TYPE_FP64) { @@ -574,7 +573,8 @@ int SubgraphTraverser::allocateTensor(std::string name) } else { - tensor->setTensorValueFloat(f16_data.size(), f16_data.data()); + std::vector<float> f32_data(f16_data.begin(), f16_data.end()); + tensor->setTensorValueFloat(f32_data.size(), f32_data.data()); } } break; diff --git a/thirdparty/serialization_lib b/thirdparty/serialization_lib -Subproject 81db8ee8f580d30ec0ca53067df32ef046e6f09 +Subproject 758e73e117c5cef17f8f0b1c543efc1df953b2f |