def sr_len_g

bend-ml-llama@0.2.1.0 · main.bend:730

import bend-ml-llama@0.2.1.0/main.bend as BendMlLlama

import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

730def sr_len_g(+t: Nat, g: TA.G) -> {sr_len(sr_read(t, g)) == 1n+t : Nat}:
731  match g:
732    case TA.G{qa, k, s}:
733      TA.zero_out(sr_len(sr_of(k, TA.read_l(t, U32.from_nat(t), Nil{}, Array.get(F32, s, U32.from_nat(t))))), 1n+t, read_len(k, t, U32.from_nat(t), Nil{}, Array.get(F32, s, U32.from_nat(t))))

Uses (12)

Used by (1)

bend-ml-llama@0.2.1.0/main:causal