gemma.cpp/backprop
Kan Wu cca75c5c60 Add configurables for norm/rope/activation/scale/residual connection.
PiperOrigin-RevId: 648971168
2024-07-03 03:04:47 -07:00
..
activations.h Prep for sharding gemma.cc: split into kv_cache, tokenizer. 2024-07-02 09:31:06 -07:00
backward-inl.h Add configurables for norm/rope/activation/scale/residual connection. 2024-07-03 03:04:47 -07:00
backward.cc Prep for sharding gemma.cc: split into kv_cache, tokenizer. 2024-07-02 09:31:06 -07:00
backward.h Move raw_weights into separate header, used mainly by compress_weights. 2024-06-17 06:17:02 -07:00
backward_scalar.h Prep for sharding gemma.cc: split into kv_cache, tokenizer. 2024-07-02 09:31:06 -07:00
backward_scalar_test.cc Add configurables for norm/rope/activation/scale/residual connection. 2024-07-03 03:04:47 -07:00
backward_test.cc Add configurables for norm/rope/activation/scale/residual connection. 2024-07-03 03:04:47 -07:00
common_scalar.h Merge pull request #212 from szabadka:adam2 2024-06-07 02:25:18 -07:00
forward-inl.h Add configurables for norm/rope/activation/scale/residual connection. 2024-07-03 03:04:47 -07:00
forward.cc Prep for sharding gemma.cc: split into kv_cache, tokenizer. 2024-07-02 09:31:06 -07:00
forward.h Add config for att/final cap, skip max-subtract. Fixes #278 2024-07-01 09:45:26 -07:00
forward_scalar.h Prep for sharding gemma.cc: split into kv_cache, tokenizer. 2024-07-02 09:31:06 -07:00
optimize_test.cc Prep for sharding gemma.cc: split into kv_cache, tokenizer. 2024-07-02 09:31:06 -07:00
optimizer.cc Use Loader/AppArgs to construct gemma_test model, simplify AcceptFunc 2024-06-11 05:53:10 -07:00
optimizer.h Support all weight types in a single binary. 2024-06-07 09:04:45 -07:00
prompt.h Add missing include 2024-06-04 10:29:12 +00:00
sampler.h Add config for att/final cap, skip max-subtract. Fixes #278 2024-07-01 09:45:26 -07:00
test_util.h Add config for att/final cap, skip max-subtract. Fixes #278 2024-07-01 09:45:26 -07:00