Searched refs:convert_boolean_attn_mask (Results 1 – 2 of 2) sorted by relevance
109 at::Tensor convert_boolean_attn_mask( in convert_boolean_attn_mask() function147 attn_mask_sliced = convert_boolean_attn_mask(attn_mask_sliced, q.dtype()); in construct_attention_mask()
516 std::optional<Tensor> convert_boolean_attn_mask(const std::optional<Tensor>& attn_mask, caffe2::Typ… in convert_boolean_attn_mask() function664 std::optional<Tensor> attn_mask = convert_boolean_attn_mask(attn_mask_, query_.dtype()); in scaled_dot_product_attention()743 attn_mask = convert_boolean_attn_mask(attn_mask, query.dtype()); in _scaled_dot_product_attention_math()