xla
Make scan-based GRU support `batch_first` parameter.
#8964

Merged

Make scan-based GRU support `batch_first` parameter. #8964

tengyifei merged 1 commit into pytorch:master from iwknow:batch

Make scan-based GRU support `batch_first` parameter.

fb05a438

tengyifei requested a review from

tengyifei 1 year ago

tengyifei approved these changes on 2025-04-11

tengyifei merged aad87e61 into master 1 year ago

iwknow deleted the batch branch 1 year ago

Reviewers

tengyifei

Assignees

No one assigned

Labels

None yet

Milestone

No milestone