def cast_f

bendtorch@1.3.0.0 · dtype.bend:623

import Base
import bendtorch@1.3.0.0/main.bend as T
import bend-ml-tensor-array@0.1.18.0/main.bend as TA

623def cast_f(-P: Type, -S: Type, +r: Nat, +c: Nat, f: T.Fwd<P, TA.Mat<r, c>, S>) -> T.Fwd<P, TA.Mat<r, c>, S>:
624  match f:
625    case T.Fwd{p, y, s}:
626      T.Fwd{p, bf16_mat(r, c, y), s}