From cc7ae5e6d1caa411a92dc6c9a2b4bbf8575f7ce9 Mon Sep 17 00:00:00 2001 From: Patryk Elszkowski Date: Thu, 22 Apr 2021 17:41:41 +0200 Subject: [PATCH] add for serialization names for i4 u4 (#5337) Co-authored-by: Patryk Elszkowski --- .../src/transformations/src/transformations/serialize.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/inference-engine/src/transformations/src/transformations/serialize.cpp b/inference-engine/src/transformations/src/transformations/serialize.cpp index c2eb4d6a5bd..873eea32a14 100644 --- a/inference-engine/src/transformations/src/transformations/serialize.cpp +++ b/inference-engine/src/transformations/src/transformations/serialize.cpp @@ -492,6 +492,8 @@ std::string get_output_precision_name(ngraph::Output& o) { return "BF16"; case ::ngraph::element::Type_t::f64: return "FP64"; + case ::ngraph::element::Type_t::i4: + return "I4"; case ::ngraph::element::Type_t::i8: return "I8"; case ::ngraph::element::Type_t::i16: @@ -500,6 +502,8 @@ std::string get_output_precision_name(ngraph::Output& o) { return "I32"; case ::ngraph::element::Type_t::i64: return "I64"; + case ::ngraph::element::Type_t::u4: + return "U4"; case ::ngraph::element::Type_t::u8: return "U8"; case ::ngraph::element::Type_t::u16: