oa::FnMatrix::gruCellLinear

gruCellLinear: fused GRU recurrent step — Linear(h, W_hh) + gruCellPointwise. Replaces the per-timestep pair of dispatches with one kernel. The hidden projection required by reverse mode is retained internally rather than exposed as an output parameter.

FnC++ public headers

Function Documentation

Matrix oa::FnMatrix::gruCellLinear(
	const Matrix & inGatesI,
	const Matrix & inHidden,
	const Matrix & inWeightHh,
	oa::I32 inHiddenSize,
	oa::U32 inTimeOffset,
	oa::U32 inBatchStride,
	const Matrix & inBiasHh = Matrix{}
)

Parameters

inGatesI

const Matrix &

—

inHidden

const Matrix &

—

inWeightHh

const Matrix &

—

inHiddenSize

oa::I32

—

inTimeOffset

oa::U32

—

inBatchStride

oa::U32

—

inBiasHh

const Matrix &

Default: Matrix{}

Returns

Matrix

The declared return value.