Searched defs:seq_mask (Results 1 – 4 of 4) sorted by relevance
87 uint64_t reconstruct_seqnum(uint16_t wire_seq, uint64_t seq_mask, in reconstruct_seqnum()
189 uint64_t reconstruct_seqnum(uint16_t wire_seq, uint64_t seq_mask, in reconstruct_seqnum()
3546 const auto seq_mask = torch::randint(0, 2, {1, seq_len}); in _multihead_attn_test_helper() local