xla
b52b4a4c - [SPMD] Output param sharding (#5018)

Commit

2 years ago

[SPMD] Output param sharding (#5018) * [SPMD] Allow non-replicated outputs in computation results. * Disable teset_dynamo_spmd

References

Author

yeounoh

Parents