Commit c89abac0 authored by Diego Caballero's avatar Diego Caballero Committed by Scott Cyphers

[ONNX] Fix support for float16 in ONNX frontend. (#3563)

Initial support for float16 was done before the ngraph::float16 data
type was introduced in nGraph. This PR aligns ONNX frontend with the
latest float16 implementation.
parent e5330c16
...@@ -48,7 +48,7 @@ namespace ngraph ...@@ -48,7 +48,7 @@ namespace ngraph
inline std::shared_ptr<ngraph::op::Constant> inline std::shared_ptr<ngraph::op::Constant>
make_ng_constant<Tensor::Type::float16>(const Tensor& tensor) make_ng_constant<Tensor::Type::float16>(const Tensor& tensor)
{ {
return __make_ng_constant<float>(element::f32, tensor); return __make_ng_constant<ngraph::float16>(element::f16, tensor);
} }
template <> template <>
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment