Skip to content

Fix vmap axes for input_positions in dot_product_attention_with_rope - #5537

Open
TheSaiEaranti wants to merge 1 commit into
google:mainfrom
TheSaiEaranti:fix-rope-input-positions
Open

Fix vmap axes for input_positions in dot_product_attention_with_rope#5537
TheSaiEaranti wants to merge 1 commit into
google:mainfrom
TheSaiEaranti:fix-rope-input-positions

Fix vmap axes for input_positions in dot_product_attention_with_rope

f2f0bb7
Select commit
Loading
Failed to load commit list.
Sign in for the full log view
check-changes
succeeded Jul 26, 2026 in 8s