Tri Dao
|
36587c01cb
[LayerNorm] Update layer_norm_linear
|
пре 9 месеци |
Tri Dao
|
bdcae547c7
[LayerNorm] Don't exit early in the backward pass (fix #781)
|
пре 11 месеци |
Tri Dao
|
c9861a032d
[LayerNorm] Initialize mean and rstd tensor using x.device
|
пре 1 година |
Tri Dao
|
f5b308e258
[LayerNorm] Rename layernorm.py -> layer_norm.py
|
пре 1 година |