DistilBERT - Operation reordering for compatibility with TensorRT
This change avoids an `expand_as` operation on a boolean tensor. Improves compatibility with TensorRT 8.0.

@LysandreJik
@mfuntowicz


Many thanks @novatig 🙏🏻 💪🏻 

As part of Infinity, this PR enables exporting DistilBERT based models to TensorRT plans for efficient inference as explained by @novatig <|||||>Test failure unrelated and fixed on `master`.<|||||>This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the [contributing guidelines](https://github.com/huggingface/transformers/blob/master/CONTRIBUTING.md) are likely to be ignored.