Skip to main content
Log in

Mojo module

mha_mask

Aliases

MASK_VALUE

alias MASK_VALUE = -10000

Structs

Traits

  • MHAMask: The MHAMask trait describes masks for MHA kernels, such as the causal mask.

Functions