Searched defs:CausalVariant (Results 1 – 2 of 2) sorted by relevance
33 class CausalVariant(IntEnum): class113 def __init__(self, variant: CausalVariant, seq_len_q: int, seq_len_kv: int):
3474 def test_causal_variants(self, device, causal_variant: CausalVariant, shape: List[Tuple[int]]):3505 …def test_causal_variants_compile(self, device, causal_variant: CausalVariant, shape: List[Tuple[in…