Function flashBackward

flashBackward(
    q: MlxArray,
    k: MlxArray,
    v: MlxArray,
    O: MlxArray,
    L: MlxArray,
    dO: MlxArray,
    scale: number,
    causal: boolean,
    window?: number,
): [MlxArray, MlxArray, MlxArray]
Three-dispatch Metal backward. q,k,v,O,dO fp16; L fp32. Returns [dQ,dK,dV].
Parameters
- q: MlxArray
- k: MlxArray
- v: MlxArray
- O: MlxArray
- L: MlxArray
- dO: MlxArray
- scale: number
- causal: boolean
- window: number = 0
Returns [MlxArray, MlxArray, MlxArray]
- Defined in packages/inference/src/kernels/attention/flash.ts:510