Searched defs:attention_mask (Results 1 – 2 of 2) sorted by relevance
417 def _masked_softmax(self, attention_scores, attention_mask=None): argument433 attention_mask=None, argument480 attention_mask=None, argument
254 attention_mask=None, argument