Add out_sharding arguments to linear layers where supported #5156
jackopenn
wants to merge
2 commits into
google:main
from
jackopenn:conv-out-shardings
Add out_sharding arguments to linear layers where supported
09a2c866
samanklesaria
approved these changes
on 2025-12-29
fix: update expected shape in conv sharding test
6c8b32e6
samanklesaria
approved these changes
on 2025-12-29
Login to write a write a comment.
Login via GitHub