Class CodebookMmaGateUp

1..8 activation rows on the simdgroup matrix unit, each code value read from a threadgroup codebook (MmaGateUp's bits); widths may differ.

Index

Constructors

Properties

Methods

Constructors

constructor

new CodebookMmaGateUp(gate: TrellisLinear, up: TrellisLinear): CodebookMmaGateUp
Parameters
- gate: TrellisLinear
- up: TrellisLinear
Returns CodebookMmaGateUp
- Defined in packages/inference/src/layers/trellis-gate-up.ts:64

Properties

`Readonly`gate

gate: TrellisLinear

`Readonly`up

up: TrellisLinear

Methods

forward

forward(x: MlxArray): MlxArray
Parameters
- x: MlxArray
Returns MlxArray
- Defined in packages/inference/src/layers/trellis-gate-up.ts:70