solve

Explicit triangular inverse policies for differentiable prefill emulation.

Functions

triangular_inverse

Materialize the exact unit-lower triangular inverse for WY operand QDQ.

triangular_inverse(lower, policy)

Materialize the exact unit-lower triangular inverse for WY operand QDQ.

Parameters:

• lower (Tensor)

• policy (LinearAttentionSolveConfig)

Return type:

Tensor