Searched defs:tgt_key_padding_mask (Results 1 – 2 of 2) sorted by relevance
169 const Tensor& tgt_key_padding_mask, in forward()334 const Tensor& tgt_key_padding_mask, in forward()427 const Tensor& tgt_key_padding_mask, in forward()
1459 torch::Tensor tgt_key_padding_mask = in transformer_test_helper() local