Searched defs:enable_gqa (Results 1 – 4 of 4) sorted by relevance
51 bool enable_gqa; member
131 …nsor> & attn_mask, double dropout_p, bool is_causal, std::optional<double> scale, bool enable_gqa){ in _fused_sdp_choice_privateuse1()
43 def create_attention(score_mod, block_mask, enable_gqa=False): argument
1996 bool enable_gqa) { in initModule()