Public
  1. Public

onnxruntime

Public

Commits

Patrice VignolaGitHub
65575389f4d
Patrice Vignola authored and GitHub committed 61f1a16265a
Fix MHA shape inference (#18009)

The previous shape inference never had the chance to infer the past_key
and past_value outputs because we were returning early.