Fix GQA shape inference (#18723)

Commit

2 years ago

Fix GQA shape inference (#18723) The shape inference is always returning before getting the chance to infer the key/value outputs.