Searched defs:key_layer (Results 1 – 2 of 2) sorted by relevance
1188 self, query_layer, key_layer, value_layer, mask=None, is_causal=True argument
862 def forward(self, query_layer, key_layer, value_layer, attn_mask): argument