Searched defs:attn_mask_dim (Results 1 – 1 of 1) sorted by relevance
270 …def test_multiheadattention_fastpath_attn_mask(self, device, attn_mask_dim, key_padding_mask_dim, … argument969 …def test_scaled_dot_product_attention(self, device, input_dim, attn_mask_dim, is_causal, dropout_p… argument