flax
53fbca51 - fix: update expected shape in conv sharding test

Commit

186 days ago

fix: update expected shape in conv sharding test

References

#5156 - Add out_sharding arguments to linear layers where supported

Author

jackopenn

jackopenn

Committer

samanklesaria

samanklesaria

Parents

Loading