Searched defs:is_dropout (Results 1 – 3 of 3) sorted by relevance
256 bool is_dropout = params.dropout > 0.0; in check_requires_grad_and_head_dim_gt192_constraints_on_sm86_89() local
839 bool is_dropout = p_dropout > 0.0; in mha_bwd() local1056 bool is_dropout = p_dropout > 0.0; in mha_varlen_bwd() local
1703 def _get_block_size_n(device, head_dim, is_dropout, is_causal): argument2020 is_dropout=False, argument