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

[nGraph] Enable u1 data type in Constant PyAPI (#2084)

Browse Source
This commit is contained in:
Jan Iwaszkiewicz 2020-09-08 13:11:05 +02:00 committed by GitHub
parent 8d74accf52
commit f4cb425396
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ngraph/core/src/type/element_type.cpp
View File

@ -357,6 +357,7 @@ namespace ngraph
{"i16", element::Type_t::i16},
{"i32", element::Type_t::i32},
{"i64", element::Type_t::i64},
{"u1", element::Type_t::u1},
{"u8", element::Type_t::u8},
{"u16", element::Type_t::u16},
{"u32", element::Type_t::u32},

4
ngraph/python/src/pyngraph/ops/constant.cpp
View File

@ -149,7 +149,7 @@ void regclass_pyngraph_op_Constant(py::module m)
{
return _cast_vector<int64_t>(self);
}
else if (element_type == ngraph::element::u8)
else if (element_type == ngraph::element::u8 || element_type == ngraph::element::u1)
{
return _cast_vector<uint8_t>(self);
}
@ -206,7 +206,7 @@ void regclass_pyngraph_op_Constant(py::module m)
{
return _get_buffer_info<int64_t>(self);
}
else if (element_type == ngraph::element::u8)
else if (element_type == ngraph::element::u8 || element_type == ngraph::element::u1)
{
return _get_buffer_info<uint8_t>(self);
}

1
ngraph/python/src/pyngraph/types/element_type.cpp
View File

@ -35,6 +35,7 @@ void regclass_pyngraph_Type(py::module m)
type.attr("i16") = ngraph::element::i16;
type.attr("i32") = ngraph::element::i32;
type.attr("i64") = ngraph::element::i64;
type.attr("u1") = ngraph::element::u1;
type.attr("u8") = ngraph::element::u8;
type.attr("u16") = ngraph::element::u16;
type.attr("u32") = ngraph::element::u32;