Fix num splits bug (#21899)
### Description
Found a bug with num splits where the heuristic isn't being performed
properly due to incorrect passing of sequence length to heuristic
function.
### Motivation and Context
We were experiencing significant performance issues with long sequence
length with flash attention due to this misconfiguration.
