PR #5212 Fixed mask sharding if inputs is sharded

Fixed mask sharding if inputs is sharded #5212

copybara-service merged 1 commit into main from fix-dropout-5209

gemini-code-assist commented on 2026-01-28

Fixed mask sharding if inputs is sharded

0bff68cf

vfdev-5 force pushed from 213e41e7 to 0bff68cf 41 days ago

vfdev-5 requested a review from

samanklesaria 41 days ago

vfdev-5 requested a review from

cgarciae 41 days ago

vfdev-5 marked this pull request as ready for review 41 days ago

samanklesaria approved these changes on 2026-01-29

cgarciae approved these changes on 2026-02-02

cgarciae added pull ready

copybara-service merged be338919 into main 36 days ago

copybara-service deleted the fix-dropout-5209 branch 36 days ago

Reviewers

cgarciae

samanklesaria

gemini-code-assist

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone