Uses of Record Class
io.github.kirstenali.deepj.tensor.AttentionGradients
Packages that use AttentionGradients
-
Uses of AttentionGradients in io.github.kirstenali.deepj.tensor
Methods in io.github.kirstenali.deepj.tensor that return AttentionGradientsModifier and TypeMethodDescriptionCausalAttentionBackend.causalAttentionBackward(Tensor query, Tensor key, Tensor value, CausalAttentionResult saved, Tensor gradient, int heads, float scale) default AttentionGradientsCausalAttentionBackend.causalAttentionBackward(Tensor query, Tensor key, Tensor value, CausalAttentionResult saved, Tensor gradient, int queryHeads, int keyValueHeads, float scale)