Searched defs:MultiheadAttention (Results 1 – 5 of 5) sorted by relevance
14 class MultiheadAttention(nn.MultiheadAttention): class
228 class MultiheadAttention(torch.ao.nn.quantizable.MultiheadAttention): class
973 class MultiheadAttention(Module): class
682 class MultiheadAttention(torch.nn.Module): class
3777 TEST_F(ModulesTest, MultiheadAttention) { in TEST_F() argument