Searched refs:num_embedding (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/ |
| D | test_expanded_weights.py | 408 def _test_embedding_model(self, model, num_embedding, device): argument
410 input = torch.randint(0, num_embedding, (batch_size, 5, 5), device=device)
412 partial(model, num_embedding=num_embedding), batch_size, input, device
590 def embedding_model(num_classes, num_embedding): argument
592 nn.Embedding(num_embedding, 15),
|