[Gemma] Fix eager attention (#29187)

Commit

1 year ago

[Gemma] Fix eager attention (#29187) * fix modelling code * add tests * fix tests * add some logit tests * style * fix fix