optimum
3460335e - Donut iobinding (#1209)

Commit

1 year ago

Donut iobinding (#1209) * added utility to compute encoder_sequence_length * style * added vit's encoder known output shapes * revised encoder output sequence formulas * added donut tests * style * corrected swin formula * added io tests * add donut to tasks manager * added tiny donut to exporters tests

References

#1209 - Donut iobinding

Author

IlyasMoutawwakil

Parents

0c399f9b

Files5

optimum
- exporters
  - tasks.py
- onnxruntime
  - modeling_seq2seq.py
tests
- exporters
  - exporters_utils.py
- onnxruntime
  - test_modeling.py
  - utils_onnxruntime_tests.py

optimum 3460335e - Donut iobinding (#1209)

optimum
3460335e - Donut iobinding (#1209)