flax
59c6c30c - Add missing mask argument to LayerNorm, RMSNorm, and GroupNorm.

Commit

2 years ago

Add missing mask argument to LayerNorm, RMSNorm, and GroupNorm.

References

#3510 - Add missing mask argument to LayerNorm, RMSNorm, and GroupNorm.

Author

carlosgmartin

carlosgmartin

Parents

Loading