def Vec4.argmax.two
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
111def Vec4.argmax.two(ix: U32 & F32, jy: U32 & F32) -> U32 & F32:
112 (i, x) = ix
113 (j, y) = jy
114 +x2 = x
115 +y2 = y
116 Vec4.argmax.pick((x2 >= y2 : F32), i, x2, j, y2)