TransposeSinking: add support for Slice and Reshape ops (#16208)
* Resolve the performance issues in TransposeSinking transformation * codestyle * fix warning as error, fix tests failures * fix ts for Concat and Reduce * Fix TransposeReduceBackward * fix the issue in TransposeFuse transformation * fix TransposeReduce transformations * Fix TransposeReduction, fix TransposeSinkingSplit, add unsqueeze support * delete debug print * Add additional validations * fix node validation * Fix validate for split, revert changes for concat, add BatchToSpace/SpaceToBatch * Add SpaceToBatch/BatchToSpace * fix TS for Interpolate + codestyle * fix gna build * Support TS for Interpolate, VariadicSplit, IsInf, IsNan, IsFinite + refactoring * add the missed line * add include * TransposeSinking tests refactoring: part1 * TransposeSinking tests refactoring: part2 * Add limited support for StridedSlice op * codestye * TransposeReduction: skip the case when 2nd input for Squeeze is not provided * Transpose sinking tests refactoring: part 3. + Revert changes in MOC. * fix build * codestyle * Add tests for TS backward transformations, update TransposeSinkingFuse transformation, delete StridedSlice transformation prototype + tests refactoring * fix unary tests * Fix warning as error on Windows * Add new tests for Unsqueeze/Squeeze; refactoring; remove debug code * TransposeSinking: add support for Slice op * Add descriptions to the transformations, add additional checks * fix a warning * TransposeSinking Rafactoring part2: move the transformations to a separate folder, align namespaces * TransposeSinking refactoring: class names, namespaces * codestyle * resolve merge conflicts * codestyle * TSReduction refactoring, move Unsqueeze/Squeeze transformations to separate files, added limited support for Reshape op + tests * fix minor mistakes * fix warnings * Added TSSlice transformation to TSGeneral, created TransposeSinkingGeneral alias in ov::pass namespace * refactoring * codestyle * fix TSSqueeze/TSUnsqueeze transformations * delete debug serialize * remove TransposeSinking from MOC * fix TSSqueeze/TSUnsqueeze transformations in case of Reshape op * delete debug code * codestyle * fix unit tests, revert changes for TSSlice transformation * fix TSSqueeze transformation * resolve review comments * codestyle
This commit is contained in:
1 parent
c5b348dd4f
commit
5a8a195dad
16 files changed
+1169
-172
No files matched your search
@@ -281,7 +281,7 @@ void FrontEnd::normalize(const std::shared_ptr<ov::Model>& model) const {
|
||||
{
|
||||
// perform transpose sinking and reverse infer if the model contains only OpenVINO operations
|
||||
ov::pass::Manager manager;
|
||||
manager.register_pass<ov::pass::transpose_sinking::TSGeneral>();
|
||||
manager.register_pass<ov::pass::TransposeSinkingGeneral>();
|
||||
manager.register_pass<ov::pass::ReverseShapeAndTypeInfer>();
|
||||
manager.run_passes(model);
|
||||
}
|
||||
|
||||
@@ -268,7 +268,7 @@ void FrontEnd::normalize(const std::shared_ptr<ov::Model>& function) const {
|
||||
manager.register_pass<ov::frontend::tensorflow_lite::pass::TFLQuantizeResolver>();
|
||||
manager.register_pass<ov::frontend::tensorflow_lite::pass::Rfft2dSimplifier>();
|
||||
manager.register_pass<ov::pass::TransposeSinking>();
|
||||
manager.register_pass<ov::pass::transpose_sinking::TSGeneral>();
|
||||
manager.register_pass<ov::pass::TransposeSinkingGeneral>();
|
||||
manager.run_passes(function);
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user