4 Commits

Author	SHA1	Message	Date
Tri Dao	66a127aef8	Refactor masking in fwd pass into 1 object	2024-01-20 17:39:53 -08:00
Tri Dao	ed4959b2eb	Change inline to __forceinline__, use __grid_constant__ param	2024-01-20 17:38:47 -08:00
Tri Dao	6777336a1c	Move masking to a separate file (mask.h)	2024-01-14 12:43:47 -08:00
Tri Dao	1274ec3e7e	Move dropout to a separate file (dropout.h)	2024-01-14 12:19:17 -08:00