Julien Plu b82fe7d258 Replace strided slice with tf.expand_dims (#10078) ...

* Replace tf.newaxis -> tf.expand_dims

* Fix tests

* Fix tests

* Use reshape when a tensors needs a double expand

* Fix GPT2

* Fix GPT2

2021-02-09 11:48:28 -05:00

..

__init__.py

Fast imports part 3 (#9474)

2021-01-08 07:40:59 -05:00

configuration_roberta.py

Reorganize repo (#8580)

2020-11-16 21:43:42 -05:00

convert_roberta_original_pytorch_checkpoint_to_pytorch.py

Fix imports in conversion scripts (#9674)

2021-01-19 09:40:15 -05:00

modeling_flax_roberta.py

[Flax] Align FlaxBertForMaskedLM with BertForMaskedLM, implement from_pretrained, init (#9054)

2020-12-16 13:03:32 +01:00

modeling_roberta.py

Fix model templates and use less than 119 chars (#9684)

2021-01-19 17:11:22 -05:00

modeling_tf_roberta.py

Replace strided slice with tf.expand_dims (#10078)

2021-02-09 11:48:28 -05:00

tokenization_roberta_fast.py

[Tokenizer Utils Base] Make pad function more flexible (#9928)

2021-02-02 10:35:27 +03:00

tokenization_roberta.py

[Tokenizer Utils Base] Make pad function more flexible (#9928)

2021-02-02 10:35:27 +03:00