def castp_b2

bendtorch@1.3.0.0 · dtype.bend:647

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

647def castp_b2(-X: Type, +r: Nat, +c: Nat, w: TA.Mat<r, c>, b: T.Bwd<TA.Mat<r, c>, X>) -> T.Bwd<TA.Mat<r, c>, X>:
648  match b:
649    case T.Bwd{wc, dx, g}:
650      T.Bwd{w, dx, bf16_mat(r, c, g)}