xla
abccaab4 - Apply gradient checkpointing to spmd ResNet benchmarking script (#4816)

Commit

2 years ago

Apply gradient checkpointing to spmd ResNet benchmarking script (#4816) Apply gradient checkpointing to spmd test scripts

References

Author

yeounoh

Parents