xla
36310efe - Fix the error where mark_step does not materalize tensors on SPMD:0

Commit
2 years ago
Fix the error where mark_step does not materalize tensors on SPMD:0
Author
Committer
Parents
Loading