Searched refs:size_source (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/ |
| D | test_torch.py | 6877 def helper(dim, dtype, device, size_result, size_source): argument
6879 index = torch.randint(0, size_result[dim], (size_source[dim],),
6882 source = torch.rand(size_source, dtype=dtype, device=device)
6884 source = torch.randint(-2, 5, size_source, dtype=dtype, device=device)
6886 source = torch.randint(0, 5, size_source, dtype=dtype, device=device)
|