[nGraph] Return Node's operation version (#6589)
* Add python test * expose node version to python api
This commit is contained in:
parent
7b24ff481f
commit
01eebba54b
@ -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);
|
||||
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user