~/jendhub

def max_step

bendtorch-core@0.1.0.0 · src/autograd/loss.bend:74

import bendtorch-core@0.1.0.0/src/autograd/loss.bend as Loss
import Base

74# PyTorch's maximum: NaN when either word is NaN, else the larger word, and
75# on a tie the later word, as `_mm512_max_ps(m, v)` keeps its second operand.
76def max_step(+m: F32, +v: F32) -> F32:
77  Bool.pick(F32, Bool.or(F32.is_gt(m, v), F32.is_ne(m, m)), m, v)

Uses (5)

Used by (1)