transformers

630adb2b - Avoid accessing shape and slices by using flatten in merge_heads