def sr0
import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorForkimport Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA
340def sr0(+mm1: Nat, +o: U32, +label: Nat, +nf: F32, r: Array<F32> & F32) -> TA.L2:
341 match r:
342 case Tuple{c, +x0}:
343 sr1(mm1, o, label, nf, TA.cea_max(mm1, o, x0, Array.get(F32, c, o)))