def argmax.go

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

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

import Base

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

Uses (6)

Used by (1)

bend-ml-llama@0.2.1.0/main:argmax