Uses of Record Class
io.github.kirstenali.deepj.tensor.CausalAttentionResult

Packages that use CausalAttentionResult

Package	Description
io.github.kirstenali.deepj.tensor

Uses of CausalAttentionResult in io.github.kirstenali.deepj.tensor

Methods in io.github.kirstenali.deepj.tensor that return CausalAttentionResult

Modifier and Type	Method	Description
CausalAttentionResult	CausalAttentionBackend.causalAttention(Tensor query, Tensor key, Tensor value, int heads, float scale)
default CausalAttentionResult	CausalAttentionBackend.causalAttention(Tensor query, Tensor key, Tensor value, int queryHeads, int keyValueHeads, float scale)

Methods in io.github.kirstenali.deepj.tensor with parameters of type CausalAttentionResult

Modifier and Type	Method	Description
AttentionGradients	CausalAttentionBackend.causalAttentionBackward(Tensor query, Tensor key, Tensor value, CausalAttentionResult saved, Tensor gradient, int heads, float scale)
default AttentionGradients	CausalAttentionBackend.causalAttentionBackward(Tensor query, Tensor key, Tensor value, CausalAttentionResult saved, Tensor gradient, int queryHeads, int keyValueHeads, float scale)