Class AneTrellisExpandReduce

Dense axis-1 expansion queued with the ANE's GPU share, without an intermediate evaluation that would serialize gate, up and down.

Index

Constructors

Properties

Methods

Constructors

constructor

new AneTrellisExpandReduce(linear: TrellisLinear): AneTrellisExpandReduce
Parameters
- linear: TrellisLinear
Returns AneTrellisExpandReduce
- Defined in packages/inference/src/layers/ane-prefill-split.ts:59

Properties

`Readonly`linear

linear: TrellisLinear

Methods

forward

forward(x: MlxArray): MlxArray
Parameters
- x: MlxArray
Returns MlxArray
- Defined in packages/inference/src/layers/ane-prefill-split.ts:60