Fix domain sharding setting for tuple instructions.
We were erroneously propagationg the entire domain sharding for each tuple operand, instead of propagating the operand subsharding. PiperOrigin-RevId: 208048503
Loading
Please sign in to comment