Joao Gante 8406fa6dd5 Add TFSpeech2Text (#15113) ...

* Add wrapper classes

* convert inner layers to tf

* Add TF Encoder and Decoder layers

* TFSpeech2Text models

* Loadable model

* TF model with same outputs as PT model

* test skeleton

* correct tests and run the fixup

* correct attention expansion

* TFSpeech2Text pask_key_values with TF format

2022-02-08 16:27:23 +00:00

5.7 KiB

Raw Blame History

View Raw