Files
openvino/model-optimizer/extensions
Svetlana Dolinina ae6b16c084 Add outputs for nodes from loops (#7844)
* initial version of transformation workable for loop

* moved transformation to back + minor changes in result names

* fixed mistake: to concatenate results of all iterations Unsqueeze should be added

* added shape inference for new nodes and extend test

* added support of TensorIterator

* If support + test;
fix to save model with TI after transformation

* fix code and tests according to run with ir_reader

* added finding max internal_layer_id in sub-graph + added comments to code

* turn off transformation because it should not be used in MO scenarios

* refactor code to find out iterations count for TensorIterator

* chenged name of final result to srtucture loop1.loop2.node for path [loop1, loop2, node]

* change port number to index of added output

* return list of new nodes

* change naming of output to standard way;
return result node as output of transformation

* refactor transformation, add more comments; fine up tests

* review fixes: add more comments, refactoring of infer function, fix in iterations count calculation

* added processing of dynamic iterations count + tests

* moved iterations count calculation to TI

* fixed bug in iterations count calculation

* fix bug with adding iterations count to wrong dimension + test

* review fixes: minor renaming + fixed bug with unset stride for TI

* move logic with output record outside function calculation iteration count;
fix case with negative start/end;
fix case when division result is not integer;
added tests for such cases

* review fixes: refactoring of toerations count calculation
2021-11-12 17:02:15 +03:00
..