Commit Graph

2 Commits

Author SHA1 Message Date
Kan Wu cca75c5c60 Add configurables for norm/rope/activation/scale/residual connection.
PiperOrigin-RevId: 648971168
2024-07-03 03:04:47 -07:00
Jan Wassenberg 7d0720675f Move raw_weights into separate header, used mainly by compress_weights.
Fix warnings in backprop/* (include)

PiperOrigin-RevId: 643983136
2024-06-17 06:17:02 -07:00