def cast_bwd

bendtorch@1.3.0.0 · dtype.bend:631

import bendtorch@1.3.0.0/dtype.bend as Dtype

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

631def cast_bwd(~P: Type, ~X: Type, ~S: Type, ~r: Nat, ~c: Nat, ~l: T.Layer<P, X, TA.Mat<r, c>, S>, p: P, s: S, dy: TA.Mat<r, c>, g: P) -> T.Bwd<P, X>:
632  T.Layer.bwd(P, X, TA.Mat<r, c>, S, l, p, s, bf16_mat(r, c, dy), g)

Uses (6)

Used by (1)

bendtorch@1.3.0.0/dtype:CastOut