Skip to content

fix(nnx): raise ValueError instead of assert for q/k/v shape checks in attention#5514

Open
Sumu004 wants to merge 1 commit into
google:mainfrom
Sumu004:fix/attention-assert-under-O
Open

fix(nnx): raise ValueError instead of assert for q/k/v shape checks in attention#5514
Sumu004 wants to merge 1 commit into
google:mainfrom
Sumu004:fix/attention-assert-under-O

Commits

Commits on Jun 25, 2026