PR #2101 Add ONNX export support for `PatchTST`

Add ONNX export support for `PatchTST` #2101

xenova merged 12 commits into main from add-patchtst-onnx

Add ONNX export support for `PatchTST`

527bbdaf

Add unit test for patchtst

767c509c

Add listed support for PatchTST

e348d47f

Add ONNX export support for patchtsmixer

eeb31595

Add task=feature-extraction

e2828ffb

xenova commented on 2024-11-28

Fix ONNX compatible unfold

43c21d02

michaelbenayoun approved these changes on 2024-12-12

echarlaix approved these changes on 2024-12-13

Formatting

29f3960d

Correctly handle negative indexing for onnx compatible unfold

519610b2

Update tests/exporters/exporters_utils.py

b26efadd

Update optimum/exporters/tasks.py

f0190332

Move dummy patch tst input generator to input_generators.py

c7f1371e

Code formatting

a054b028

xenova merged b9fa9aa8 into main 364 days ago

xenova deleted the add-patchtst-onnx branch 364 days ago

Reviewers

echarlaix

michaelbenayoun

Assignees

No one assigned

Labels

None yet

Milestone

No milestone