IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Enable read onnx model tensor FLOAT16 no raw data (#1168)

Browse Source
This commit is contained in:
Katarzyna Mitrus
2020-07-03 13:45:00 +02:00
committed by GitHub
parent 38335b1883
commit 099edd2f67
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
ngraph/src/ngraph/frontend/onnx_import/core/tensor.hpp
View File

@@ -215,11 +215,16 @@ namespace ngraph
inline std::vector<ngraph::float16>
get_data(const ONNX_NAMESPACE::TensorProto& tensor)
{
NGRAPH_CHECK(tensor.data_type() == ONNX_NAMESPACE::TensorProto_DataType_FLOAT16,
"Expected FLOAT16 data type");
NGRAPH_CHECK(tensor.has_raw_data(), "Expected raw data for FLOAT16 data type");
return detail::__get_raw_data<ngraph::float16>(tensor.raw_data(),
tensor.data_type());
if (tensor.has_raw_data())
{
return detail::__get_raw_data<ngraph::float16>(tensor.raw_data(),
tensor.data_type());
}
if (tensor.data_type() == ONNX_NAMESPACE::TensorProto_DataType_FLOAT16)
{
return detail::__get_data<ngraph::float16>(tensor.int32_data());
}
throw error::tensor::invalid_data_type{tensor.data_type()};
}
template <>