xla
eff1f328 - GPT-2 benchmarking - don't use SPMD:0 on PjRtShardedData

Commit
2 years ago
GPT-2 benchmarking - don't use SPMD:0 on PjRtShardedData
Author
Committer
Parents
Loading