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

Remove ngraph::Lambda class, replace TensorIterator body with ngraph::Function (#1830)

Browse Source 
* remove Lambda class, replace TensorIterator body with ngraph::Function

* Fix passing parameters from parent graph to subgraph

Co-authored-by: mbencer <mateusz.bencer@intel.com>
This commit is contained in:
Ivan Tikhonov
2020-08-19 07:09:32 +03:00
committed by GitHub
parent 618c6f7f23
commit c5ca8f5b51
24 changed files with 452 additions and 520 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ngraph/python/src/pyngraph/tensor_iterator_builder.cpp
View File

@@ -108,8 +108,7 @@ void util::TensorIteratorBuilder::get_graph_body()
m_body_outputs = as_output_vector(body_attrs["results"].cast<ngraph::NodeVector>());
m_body_parameters = body_attrs["parameters"].cast<ngraph::ParameterVector>();
m_body =
std::make_shared<ngraph::op::TensorIterator::BodyLambda>(m_body_outputs, m_body_parameters);
m_body = std::make_shared<ngraph::Function>(m_body_outputs, m_body_parameters);
}
void util::TensorIteratorBuilder::set_tensor_iterator_sliced_inputs(

2
ngraph/python/src/pyngraph/tensor_iterator_builder.hpp
View File

@@ -126,7 +126,7 @@ namespace util
const py::dict& m_attributes;
ngraph::OutputVector m_body_outputs;
ngraph::ParameterVector m_body_parameters;
std::shared_ptr<ngraph::op::TensorIterator::BodyLambda> m_body;
std::shared_ptr<ngraph::Function> m_body;
py::list m_slice_input_desc;
py::list m_merged_input_desc;
py::list m_invariant_input_desc;