Skip to content

Accept inputs shape in Attention/MLA constructor#2089

Merged
copybara-service[bot] merged 1 commit into
mainfrom
bvandermoon-nnx-attention-shape
Aug 6, 2025
Merged

Accept inputs shape in Attention/MLA constructor#2089
copybara-service[bot] merged 1 commit into
mainfrom
bvandermoon-nnx-attention-shape

Accept inputs shape in Attention/MLA constructor

dffa82b
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Aug 6, 2025 in 2s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

dffa82b Author: @bvandermoon <bva******on​@google.com>

(Only the first commit for a unique contributor is listed.)