Joao Gante e6d27ca5c8 TF: XLA beam search + most generation-compatible models are now also XLA-generate-compatible (#17857) ...

* working beam search 🎉

* XLA generation compatible with ALL classes

* add xla generation slow test

2022-06-29 12:41:01 +01:00

..

__init__.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_modeling_flax_t5.py

Black preview (#17217)

2022-05-12 16:25:55 -04:00

test_modeling_t5.py

Skip disk offload test for T5

2022-06-07 11:11:40 -04:00

test_modeling_tf_t5.py

TF: XLA beam search + most generation-compatible models are now also XLA-generate-compatible (#17857)

2022-06-29 12:41:01 +01:00

test_tokenization_t5.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00