[XLA] Rework dot() sharding propagation to lookahead instructions sharding to...
[XLA] Rework dot() sharding propagation to lookahead instructions sharding to choose a sharding for dot() that agrees with the users if possible. PiperOrigin-RevId: 565086052
Showing
想要评论请 注册 或 登录