def bf16_mat

bendtorch@1.3.0.0 · dtype.bend:620

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

620def bf16_mat(+r: Nat, +c: Nat, m: TA.Mat<r, c>) -> TA.Mat<r, c>:
621  T.Ew.map(~BF16.round, r, c, m)