Lines Matching full:embedding_bag

202         res_F = F.embedding_bag(a, embeddings)
208         res_F = F.embedding_bag(a, embeddings, padding_idx=2)
225                     F.embedding_bag(a, embeddings, padding_idx=padding_idx)
231                 F.embedding_bag(a, embeddings, padding_idx=padding_idx)
482     # Check correctness of torch.nn.functional.embedding_bag forward and
522             # embedding_bag requires first entry of offsets to be 0
601                 bag = torch.nn.functional.embedding_bag(
611                 bag_check = torch.nn.functional.embedding_bag(
635     # Check correctness of torch.nn.functional.embedding_bag forward and
642         # Use a Python implementation of embedding_bag with padding_idx support
643         # to check torch.nn.functional.embedding_bag correctness
725                 # Check forward with a Python implementation of padding_idx embedding_bag
729                 bag = torch.nn.functional.embedding_bag(
810                 torch.nn.functional.embedding_bag(
820             lambda x, y, z: torch.nn.functional.embedding_bag(y, x, z),
821             torch.embedding_bag,
1575             output_non_contig = F.embedding_bag(
1581             output_contig = F.embedding_bag(