flax
0bff68cf - Fixed mask sharding if inputs is sharded

Commit

32 days ago

Fixed mask sharding if inputs is sharded Fixes #5209

References

#5212 - Fixed mask sharding if inputs is sharded

Author

vfdev-5

vfdev-5

Committer

vfdev-5

vfdev-5

Parents

Loading