add for serialization names for i4 u4 (#5337)

Co-authored-by: Patryk Elszkowski <patryk.elszkowki@intel.com>
This commit is contained in:
Patryk Elszkowski 2021-04-22 17:41:41 +02:00 committed by GitHub
parent 636736825c
commit cc7ae5e6d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -492,6 +492,8 @@ std::string get_output_precision_name(ngraph::Output<Node>& o) {
return "BF16"; return "BF16";
case ::ngraph::element::Type_t::f64: case ::ngraph::element::Type_t::f64:
return "FP64"; return "FP64";
case ::ngraph::element::Type_t::i4:
return "I4";
case ::ngraph::element::Type_t::i8: case ::ngraph::element::Type_t::i8:
return "I8"; return "I8";
case ::ngraph::element::Type_t::i16: case ::ngraph::element::Type_t::i16:
@ -500,6 +502,8 @@ std::string get_output_precision_name(ngraph::Output<Node>& o) {
return "I32"; return "I32";
case ::ngraph::element::Type_t::i64: case ::ngraph::element::Type_t::i64:
return "I64"; return "I64";
case ::ngraph::element::Type_t::u4:
return "U4";
case ::ngraph::element::Type_t::u8: case ::ngraph::element::Type_t::u8:
return "U8"; return "U8";
case ::ngraph::element::Type_t::u16: case ::ngraph::element::Type_t::u16: