[webgpu] Apply template to flash attention (microsoft#25722)

### Description
This PR applies template to flash attention, and simplifies the
`is_unidirectional` check in shader.



### Motivation and Context
See above.

Author: daijh