Lines Matching full:at
67 arg_map["self"] = "at::rand({6, 6})"
68 arg_map["vec1"] = "at::rand({6})"
69 arg_map["vec2"] = "at::rand({6})"
71 arg_map["self"] = "at::rand({22, 22})"
72 arg_map["vec1"] = "at::rand({22})"
73 arg_map["vec2"] = "at::rand({22})"
77 arg_map["self"] = "at::rand({6, 6})"
78 arg_map["vec"] = "at::rand({6})"
80 arg_map["self"] = "at::rand({22, 22})"
81 arg_map["vec"] = "at::rand({22})"
85 arg_map["self"] = "at::rand({6, 6})"
87 arg_map["self"] = "at::rand({22, 22})"
91 arg_map["self"] = "at::rand({3, 3, 3})"
92 arg_map["other"] = "at::rand({3, 3, 3})"
94 arg_map["self"] = "at::rand({22, 3, 22})"
95 arg_map["other"] = "at::rand({22, 3, 22})"
99 arg_map["index"] = "at::randint(0, 216, {20}, torch::kInt64)"
101 arg_map["index"] = "at::randint(0, 1000, {100}, torch::kInt64)"
105 arg_map["indices"] = "at::argsort(self0, 1, true)"
107 arg_map["indices"] = "at::argsort(self1, 1, true)"
111 arg_map["mask"] = "at::randn({6, 6, 6}) > 0.5"
113 arg_map["mask"] = "at::rand({22, 22, 22}) > 0.5"
117 arg_map["input2"] = "at::rand({6, 6})"
119 arg_map["input2"] = "at::rand({22, 22})"
123 arg_map["input2"] = "at::rand({6, 6})"
125 arg_map["input2"] = "at::rand({22, 22})"
129 arg_map["q"] = "at::rand({6})"
132 arg_map["q"] = "at::rand({22})"
137 arg_map["q"] = "at::rand({6})"
140 arg_map["q"] = "at::rand({22})"
145 arg_map["self"] = "at::rand({6, 6})"
146 arg_map["target"] = "at::randint(6, {6}, torch::kInt64)"
147 arg_map["weight"] = "at::rand({6})"
149 arg_map["self"] = "at::rand({22, 22})"
150 arg_map["target"] = "at::randint(22, {22}, torch::kInt64)"
151 arg_map["weight"] = "at::rand({22})"
155 arg_map["self"] = "at::rand({6, 6})"
156 arg_map["target"] = "at::randint(6, {6, 6}, torch::kInt64)"
158 arg_map["self"] = "at::rand({22, 22})"
159 arg_map["target"] = "at::randint(22, {22, 22}, torch::kInt64)"
163 arg_map["self"] = "at::rand({6, 6})"
164 arg_map["target"] = "at::randint(6, {6}, torch::kInt64)"
165 arg_map["weight"] = "at::rand({6})"
167 arg_map["self"] = "at::rand({22, 22})"
168 arg_map["target"] = "at::randint(22, {22}, torch::kInt64)"
169 arg_map["weight"] = "at::rand({22})"
173 arg_map["self"] = "at::rand({6, 6, 6, 6})"
174 arg_map["target"] = "at::randint(6, {6, 6, 6}, torch::kInt64)"
175 arg_map["weight"] = "at::rand({6})"
177 arg_map["self"] = "at::rand({22, 22, 22, 22})"
178 arg_map["target"] = "at::randint(22, {22, 22, 22}, torch::kInt64)"
179 arg_map["weight"] = "at::rand({22})"
193 arg_map["self"] = "at::rand({6, 6, 6, 6})"
196 arg_map["self"] = "at::rand({22, 22, 22, 22})"
201 arg_map["self"] = "at::rand({2})"
202 arg_map["mat"] = "at::rand({2, 2})"
203 arg_map["vec"] = "at::rand({2})"
205 arg_map["self"] = "at::rand({35})"
206 arg_map["mat"] = "at::rand({35, 35})"
207 arg_map["vec"] = "at::rand({35})"
211 arg_map["self"] = "at::rand({2, 2, 2}) + at::ones({2, 2, 2})"
213 arg_map["self"] = "at::rand({5, 5, 5}) + at::ones({5, 5, 5})"
217 arg_map["grad_output"] = "at::rand({2, 2, 2}, at::kFloat)"
218 arg_map["self"] = "at::rand({2, 2, 2}, at::kFloat)"
219 arg_map["indices"] = "at::randint(0, 1, {2, 2, 2}, at::kLong)"
221 arg_map["grad_output"] = "at::rand({3, 3, 3}, at::kFloat)"
222 arg_map["self"] = "at::rand({3, 3, 3}, at::kFloat)"
223 arg_map["indices"] = "at::randint(0, 1, {3, 3, 3}, at::kLong)"
227 arg_map["grad_output"] = "at::rand({2, 2, 2, 2}, at::kFloat)"
228 arg_map["self"] = "at::rand({2, 2, 2, 2}, at::kFloat)"
229 arg_map["indices"] = "at::randint(0, 1, {2, 2, 2, 2}, at::kLong)"
231 arg_map["grad_output"] = "at::rand({3, 3, 3, 3}, at::kFloat)"
232 arg_map["self"] = "at::rand({3, 3, 3, 3}, at::kFloat)"
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)"
265 arg_map["self"] = "at::rand({6, 6, 6})"
268 arg_map["self"] = "at::rand({22, 22, 22})"
273 arg_map["grad_output"] = "at::rand({6, 6, 6})"
274 arg_map["self"] = "at::rand({6, 6, 6})"
277 arg_map["grad_output"] = "at::rand({22, 22, 22})"
278 arg_map["self"] = "at::rand({22, 22, 22})"
283 arg_map["self"] = "at::rand({2})"
285 arg_map["index"] = "at::randint(0, 1, {2}, at::kInt)"
286 arg_map["source"] = "at::rand({2})"
289 arg_map["self"] = "at::rand({16})"
291 arg_map["index"] = "at::randint(0, 10, {16}, at::kInt)"
292 arg_map["source"] = "at::rand({16})"
297 arg_map["self"] = "at::rand({2})"
299 arg_map["index"] = "at::randint(0, 1, {2}, at::kLong)"
300 arg_map["source"] = "at::rand({2})"
302 arg_map["self"] = "at::rand({32})"
304 arg_map["index"] = "at::randint(0, 10, {32}, at::kLong)"
305 arg_map["source"] = "at::rand({32})"
309 arg_map["self"] = "at::rand({6, 3, 6})"
310 arg_map["other"] = "at::rand({6, 3, 6})"
313 arg_map["self"] = "at::rand({22, 3, 22})"
314 arg_map["other"] = "at::rand({22, 3, 22})"
319 arg_map["grad_output"] = "at::rand({})"
320 arg_map["self"] = "at::rand({6})"
321 arg_map["target"] = "at::randint(0, 5, {6}, torch::kInt64)"
322 arg_map["weight"] = "at::rand({6})"
325 arg_map["total_weight"] = "at::rand({})"
327 arg_map["grad_output"] = "at::rand({})"
328 arg_map["self"] = "at::rand({36})"
329 arg_map["target"] = "at::randint(0, 11, {36}, torch::kInt64)"
330 arg_map["weight"] = "at::rand({36})"
333 arg_map["total_weight"] = "at::rand({})"
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)"
356 arg_map["self"] = "at::rand({2,2,2}, at::kDouble) + at::ones({2,2,2})"
357 arg_map["other"] = "at::rand({2,2,2}, at::kDouble) + at::ones({2,2,2})"
359 arg_map["self"] = "at::rand({5,5,5}, at::kDouble) + at::ones({5,5,5})"
360 arg_map["other"] = "at::rand({5,5,5}, at::kDouble) + at::ones({5,5,5})"
376 arg_map["self"] = "at::randint(0, 3, {2}, at::kInt)"
380 arg_map["self"] = "at::randint(0, 3, {12}, at::kInt)"