Searched defs:src_key_padding_mask (Results 1 – 2 of 2) sorted by relevance
66 const Tensor& src_key_padding_mask) { in forward()265 const Tensor& src_key_padding_mask) { in forward()426 const Tensor& src_key_padding_mask, in forward()
881 def forward(self, src, src_mask=None, src_key_padding_mask=None): argument