Julien Plu 83d803ba02 Making TF BART-like models XLA and AMP compliant (#10191) ...

* Update BART

* Update Blenderbot

* Update BlenderbotSmall

* Update Marian

* Update MBart

* Update MBart

* Update Pegasus

* Update template

* Fix Marian and Pegasus

* Apply style

* Default initializer

* Default initializer

* Default initializer

* Remove int32 casts

* Fix template

* Remove more cast

2021-02-17 17:48:56 +01:00

62 KiB

Raw Blame History

View Raw