Lines Matching full:randint

99             arg_map["index"] = "at::randint(0, 216, {20}, torch::kInt64)"
101             arg_map["index"] = "at::randint(0, 1000, {100}, torch::kInt64)"
146             arg_map["target"] = "at::randint(6, {6}, torch::kInt64)"
150             arg_map["target"] = "at::randint(22, {22}, torch::kInt64)"
156             arg_map["target"] = "at::randint(6, {6, 6}, torch::kInt64)"
159             arg_map["target"] = "at::randint(22, {22, 22}, torch::kInt64)"
164             arg_map["target"] = "at::randint(6, {6}, torch::kInt64)"
168             arg_map["target"] = "at::randint(22, {22}, torch::kInt64)"
174             arg_map["target"] = "at::randint(6, {6, 6, 6}, torch::kInt64)"
178             arg_map["target"] = "at::randint(22, {22, 22, 22}, torch::kInt64)"
219             arg_map["indices"] = "at::randint(0, 1, {2, 2, 2}, at::kLong)"
223             arg_map["indices"] = "at::randint(0, 1, {3, 3, 3}, at::kLong)"
229             arg_map["indices"] = "at::randint(0, 1, {2, 2, 2, 2}, at::kLong)"
233             arg_map["indices"] = "at::randint(0, 1, {3, 3, 3, 3}, at::kLong)"
237             arg_map["self"] = "at::randint(1, 1 << 4, {6, 6, 6}, at::kInt)"
238             arg_map["other"] = "at::randint(1, 26, {6, 6, 6}, at::kInt)"
240             arg_map["self"] = "at::randint(1, 1 << 4, {22, 22, 22}, at::kInt)"
241             arg_map["other"] = "at::randint(1, 26, {22, 22, 22}, at::kInt)"
245             arg_map["self"] = "at::randint(1 << 21, 1 << 30, {6, 6, 6}, at::kInt)"
246             arg_map["other"] = "at::randint(1, 22, {6, 6, 6}, at::kInt)"
248             arg_map["self"] = "at::randint(1 << 21, 1 << 30, {22, 22, 22}, at::kInt)"
249             arg_map["other"] = "at::randint(1, 22, {22, 22, 22}, at::kInt)"
253             arg_map["self"] = "at::randint(1, 100, {2,2,2}, at::kInt)"
255             arg_map["index"] = "at::randint(0, 1, {2,2,2}, torch::kInt64)"
258             arg_map["self"] = "at::randint(1, 100, {5,5,5}, at::kInt)"
260             arg_map["index"] = "at::randint(0, 4, {5,5,5}, torch::kInt64)"
285             arg_map["index"] = "at::randint(0, 1, {2}, at::kInt)"
291             arg_map["index"] = "at::randint(0, 10, {16}, at::kInt)"
299             arg_map["index"] = "at::randint(0, 1, {2}, at::kLong)"
304             arg_map["index"] = "at::randint(0, 10, {32}, at::kLong)"
321             arg_map["target"] = "at::randint(0, 5, {6}, torch::kInt64)"
329             arg_map["target"] = "at::randint(0, 11, {36}, torch::kInt64)"
337             arg_map["self"] = "at::randint(1, 100, {2,2,2}, torch::kInt64)"
338             arg_map["index"] = "at::randint(0, 1, {2,2,2}, torch::kInt64)"
339             arg_map["src"] = "at::randint(1, 100, {2,2,2}, torch::kInt64)"
341             arg_map["self"] = "at::randint(1, 100, {5,5,5}, torch::kInt64)"
342             arg_map["index"] = "at::randint(0, 1, {5,5,5}, torch::kInt64)"
343             arg_map["src"] = "at::randint(1, 100, {5,5,5}, torch::kInt64)"
350             arg_map["index"] = "at::randint(6, {6, 6, 6}, torch::kInt64)"
352             arg_map["index"] = "at::randint(22, {22, 22, 22}, torch::kInt64)"
376             arg_map["self"] = "at::randint(0, 3, {2}, at::kInt)"
380             arg_map["self"] = "at::randint(0, 3, {12}, at::kInt)"