Commit Graph

7 Commits

Author SHA1 Message Date
Maxim Vafin
8d0381b0fe [PT FE] Implement custom op for types alignment (#20431)
* [PT FE] Implement custom op for types alignment

* Fix code style

* Fix inplace ops

* Fix layer tests

* Remove no longer needed change

* Fix ovc tests

* Fix fe tests
2023-10-23 22:54:08 +02:00
Maxim Vafin
f38b5f4f06 [PT FE] Support moving TupleConstruct inside If body (#20081)
* Support moving TupleConstruct inside If body

* Fix win build

---------

Co-authored-by: Alina Kladieva <alina.kladieva@intel.com>
2023-09-28 23:11:09 +02:00
Maxim Vafin
37d54bcb42 [PT FE] Support prim::TupleIndex operation (#19978)
* [PT FE] Support prim::TupleIndex

* Update src/frontends/pytorch/src/op/tuple_index.cpp

* Update src/frontends/pytorch/src/op/tuple_index.cpp
2023-09-21 15:57:42 +02:00
Sergey Lyalin
dfba702c74 Decompose/flatten tuple inputs (#18092)
* prim::TupleUnpack and prim::ListUnpack removing transformation in PT FE to flatten input list and tuples

* Enabled tuples and lists as items in example_inputs

* Applied code style

* Added tests for tuples as inputs and extended test infrastructure to support it

* Negligible performance optimizations

Co-authored-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Fixed duplicated names of test classes

* Added description for tuple flattening transformation

* Removed any support for list flattening on inputs; fixed layer tests

* Fixed style

* Fixed order of new Parameters and Results while flattening tuples

* Fixed style

* Better diagnostics when not all prim::TupleUnpack ops after Parameters are decomposed

* Small fix in diagnostics message

---------

Co-authored-by: Ekaterina Aidova <ekaterina.aidova@intel.com>
Co-authored-by: Andrei Kochin <andrei.kochin@intel.com>
Co-authored-by: Roman Kazantsev <roman.kazantsev@intel.com>
Co-authored-by: Alina Kladieva <alina.kladieva@intel.com>
2023-07-06 11:05:26 +04:00
Ekaterina Aidova
df0bd18ed2 [PT FE]: fix constant folding for dequantization (#18190)
* [PT FE]: fix constant folding for dequantization

* add test
2023-06-23 06:41:32 +02:00
Ekaterina Aidova
cd340301d8 [PT FE]: support prim::min, prim::ListConstruct as output and fix aten::getitem for prim::ListConstruct (#15470) 2023-02-02 22:36:07 +00:00
Maxim Vafin
994b227b86 Remove None at outputs of the model, improve types handling in frontend (#15258)
* Remove None at outputs of the model, improve types handling in frontend

* Fix py code style

* Add torch dependency in pybind tests

* Fix tests if fe is disabled and add backward type cpnversion

* Move decoder tests to layer tests

* Fix codestyle

* Add comment

* Move tests to separate folder

* Update .ci/azure/linux.yml
2023-01-25 13:28:47 +03:00