gemma.cpp/backprop
Jan Wassenberg 45ad847a41 Replace RowVectorBatch with MatStorageT
KVCache: add ctor required for MatStorageT, remove Create; bf_pre_ffw_rms_out -> pre_ffw_rms_out
optimize_test: larger vocab_size requires more steps
shared.h: Remove unused u128 type
correctly set Activation matrix rows, avoid passing as arg
ops: pass Mat instead of pointers/sizes; vectorize LayerNorm; support any weight type
mat: add OverrideRows, used by SetBatchSize
PiperOrigin-RevId: 757790736
2025-05-12 09:16:12 -07:00
..
activations.h Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
backward-inl.h Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
backward.cc Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
backward.h Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
backward_scalar.h Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
backward_test.cc Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
common_scalar.h Major refactor of allocator/args: 2025-04-10 01:29:54 -07:00
forward-inl.h Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
forward.cc Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
forward.h Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
forward_scalar.h Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
optimize_test.cc Replace RowVectorBatch with MatStorageT 2025-05-12 09:16:12 -07:00
optimizer.cc Huge refactor of weight handling and model loading. 2025-05-06 04:44:21 -07:00
optimizer.h Huge refactor of weight handling and model loading. 2025-05-06 04:44:21 -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 Rename-only: remove Allocator2 etc suffixes now that refactoring is complete 2025-05-06 09:12:43 -07:00