Class AneTrellisExpandedMlpSplit

Fixed dense GPU share for written row-major ANE graph bindings.

Index

Constructors

Methods

Constructors

constructor

new AneTrellisExpandedMlpSplit(): AneTrellisExpandedMlpSplit

Returns AneTrellisExpandedMlpSplit

Methods

`Static`build

build(
    gate: TrellisLinear,
    up: TrellisLinear,
    down: TrellisLinear,
    fraction: number,
    rows: number,
    share?: AneMlpSplit,
    slices?: number,
): AneTrellisMlpSplit
Parameters
- gate: TrellisLinear
- up: TrellisLinear
- down: TrellisLinear
- fraction: number
- rows: number
- Optionalshare: AneMlpSplit
- Optionalslices: number
Returns AneTrellisMlpSplit
- Defined in packages/inference/src/layers/ane-prefill-split.ts:206