xla

3913a77e - Add API to assemble CPU shards to a sharded tensor (#5630)

Add API to assemble CPU shards to a sharded tensor (#5630) * Add API to assemble CPU shards to a sharded tensor * Handle replicated sharding * Move validations into get_op_sharding * Improve tests and error handling