xla
allow disabling layout pinning in optimizer_step and reduce_gradients
#3556
Merged

allow disabling layout pinning in optimizer_step and reduce_gradients #3556

ronghanghu
ronghanghu allow disabling layout pinning in optimizer_step and reduce_gradients
2f9ed684
JackCaoG JackCaoG requested a review from JackCaoG JackCaoG 3 years ago
JackCaoG
JackCaoG approved these changes on 2022-05-09
JackCaoG JackCaoG merged 6e3992ac into master 3 years ago
ronghanghu ronghanghu deleted the optimizer-layout-pinning branch 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone