Fix #20429 — Attention layer hardcodes 3-D shape assumptions, breaking N-D inputs
#10757
| Job | Run time |
|---|---|
| 2m 15s | |
| 9m 35s | |
| 28m 41s | |
| 25m 20s | |
| 10m 53s | |
| 47m 4s | |
| 3m 26s | |
| 2h 7m 14s |