Class ReduceSplitMixedGateUp

Fixed split activation for the written M1 graph's mixed-width projections.

Hierarchy (View Summary)

ReduceMixedGateUp
- ReduceSplitMixedGateUp

Index

Constructors

Properties

Methods

Constructors

constructor

new ReduceSplitMixedGateUp(
gate: TrellisLinear,
up: TrellisLinear,
): ReduceSplitMixedGateUp
Parameters
- gate: TrellisLinear
- up: TrellisLinear
Returns ReduceSplitMixedGateUp
Overrides ReduceMixedGateUp.constructor
- Defined in packages/inference/src/layers/trellis-reduce-gate-up.ts:72

Properties

`Readonly`gate

gate: TrellisLinear

`Readonly`tail

tail: MixedGateUpTail

`Readonly`up

up: TrellisLinear

Methods

forward

forward(x: MlxArray): MlxArray
x [.., in] (borrowed, 1..4 rows) to an owned [.., rows].
Parameters
- x: MlxArray
Returns MlxArray
Inherited from ReduceMixedGateUp.forward
- Defined in packages/inference/src/layers/trellis-reduce-gate-up.ts:65