oa::FnMatrix::swigluBwd

swigluBwd: backward pass for SwiGLU activation.

Function Documentation

SwigluBwdResult oa::FnMatrix::swigluBwd( const Matrix & inGate, const Matrix & inUp, const Matrix & inOut, const Matrix & inGradOutput )
swigluBwd: backward pass for SwiGLU activation.

Parameters

inGate
const Matrix &

inUp
const Matrix &

inOut
const Matrix &

inGradOutput
const Matrix &

Returns

SwigluBwdResult

SwigluBwdResult with dGate, dUp