[nGraph] Return Node's operation version (#6589)

* Add python test

* expose node version to python api
This commit is contained in:
Jan Iwaszkiewicz 2021-07-13 11:49:11 +02:00 committed by GitHub
parent 7b24ff481f
commit 01eebba54b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 0 deletions

View File

@ -248,12 +248,23 @@ void regclass_pyngraph_Node(py::module m)
get_rt_info : PyRTMap
A dictionary of user defined data.
)");
node.def("get_version",
&ngraph::Node::get_version,
R"(
Returns operation's version of the node.
Returns
----------
get_version : int
Operation version.
)");
node.def_property_readonly("shape", &ngraph::Node::get_shape);
node.def_property_readonly("name", &ngraph::Node::get_name);
node.def_property_readonly("rt_info",
(PyRTMap & (ngraph::Node::*)()) & ngraph::Node::get_rt_info,
py::return_value_policy::reference_internal);
node.def_property_readonly("version", &ngraph::Node::get_version);
node.def_property(
"friendly_name", &ngraph::Node::get_friendly_name, &ngraph::Node::set_friendly_name);

View File

@ -425,3 +425,10 @@ def test_sink_function_ctor():
assert len(function.get_parameters()) == 1
assert len(function.get_results()) == 1
assert function.get_friendly_name() == "TestFunction"
def test_node_version():
node = ng.add([1], [2])
assert node.get_version() == 1
assert node.version == 1