xla
Replace upstream GRU implementation with scan-based GRU
#9010
Merged

Replace upstream GRU implementation with scan-based GRU #9010

tengyifei merged 2 commits into pytorch:master from iwknow:replace
iwknow
iwknow Replace upstream GRU implementation with Scan-based GRU implementatio…
89b3dc10
iwknow Update GRU tests after the nn.GRU is replaced with scan-based impleme…
80e73a52
iwknow
tengyifei
tengyifei approved these changes on 2025-04-19
tengyifei tengyifei enabled auto-merge (squash) 236 days ago
tengyifei tengyifei merged faf777a2 into master 236 days ago
iwknow iwknow deleted the replace branch 218 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone