def argmax.go

bend-ml-llama@0.2.0.0 · main.bend:1093

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

import Base

1093def argmax.go(xs: List<&2, F32>, g: Bool, +i: Nat, +bi: Nat, +bv: F32) -> AM:
1094  match xs g:
1095    case Nil{} _:
1096      AM{bi, bv}
1097    case Con{+h, +t} True{}:
1098      argmax.go(t, gt_head(t, h), 1n+i, i, h)
1099    case Con{h, +t} False{}:
1100      argmax.go(t, gt_head(t, bv), 1n+i, bi, bv)

Uses (6)

Used by (1)

bend-ml-llama@0.2.0.0/main:argmax