1#[allow(clippy::all)]
3use crate::{C_scalar, C_tensor};
4use libc::c_int;
5
6extern "C" {
7 pub fn atg___and__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
8 pub fn atg___and__tensor_(
9 out__: *mut *mut C_tensor,
10 self_: *mut C_tensor,
11 other_: *mut C_tensor,
12 );
13 pub fn atg___iand__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
14 pub fn atg___iand__tensor_(
15 out__: *mut *mut C_tensor,
16 self_: *mut C_tensor,
17 other_: *mut C_tensor,
18 );
19 pub fn atg___ilshift__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
20 pub fn atg___ilshift__tensor_(
21 out__: *mut *mut C_tensor,
22 self_: *mut C_tensor,
23 other_: *mut C_tensor,
24 );
25 pub fn atg___ior__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
26 pub fn atg___ior__tensor_(
27 out__: *mut *mut C_tensor,
28 self_: *mut C_tensor,
29 other_: *mut C_tensor,
30 );
31 pub fn atg___irshift__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
32 pub fn atg___irshift__tensor_(
33 out__: *mut *mut C_tensor,
34 self_: *mut C_tensor,
35 other_: *mut C_tensor,
36 );
37 pub fn atg___ixor__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
38 pub fn atg___ixor__tensor_(
39 out__: *mut *mut C_tensor,
40 self_: *mut C_tensor,
41 other_: *mut C_tensor,
42 );
43 pub fn atg___lshift__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
44 pub fn atg___lshift__scalar_out_(
45 out__: *mut *mut C_tensor,
46 out_: *mut C_tensor,
47 self_: *mut C_tensor,
48 other_: *mut C_scalar,
49 );
50 pub fn atg___lshift__tensor_(
51 out__: *mut *mut C_tensor,
52 self_: *mut C_tensor,
53 other_: *mut C_tensor,
54 );
55 pub fn atg___lshift__tensor_out_(
56 out__: *mut *mut C_tensor,
57 out_: *mut C_tensor,
58 self_: *mut C_tensor,
59 other_: *mut C_tensor,
60 );
61 pub fn atg___or__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
62 pub fn atg___or__tensor_(
63 out__: *mut *mut C_tensor,
64 self_: *mut C_tensor,
65 other_: *mut C_tensor,
66 );
67 pub fn atg___rshift__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
68 pub fn atg___rshift__scalar_out_(
69 out__: *mut *mut C_tensor,
70 out_: *mut C_tensor,
71 self_: *mut C_tensor,
72 other_: *mut C_scalar,
73 );
74 pub fn atg___rshift__tensor_(
75 out__: *mut *mut C_tensor,
76 self_: *mut C_tensor,
77 other_: *mut C_tensor,
78 );
79 pub fn atg___rshift__tensor_out_(
80 out__: *mut *mut C_tensor,
81 out_: *mut C_tensor,
82 self_: *mut C_tensor,
83 other_: *mut C_tensor,
84 );
85 pub fn atg___xor__(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
86 pub fn atg___xor__tensor_(
87 out__: *mut *mut C_tensor,
88 self_: *mut C_tensor,
89 other_: *mut C_tensor,
90 );
91 pub fn atg__adaptive_avg_pool2d(
92 out__: *mut *mut C_tensor,
93 self_: *mut C_tensor,
94 output_size_data: *const i64,
95 output_size_len: c_int,
96 );
97 pub fn atg__adaptive_avg_pool2d_backward(
98 out__: *mut *mut C_tensor,
99 grad_output_: *mut C_tensor,
100 self_: *mut C_tensor,
101 );
102 pub fn atg__adaptive_avg_pool2d_backward_out(
103 out__: *mut *mut C_tensor,
104 out_: *mut C_tensor,
105 grad_output_: *mut C_tensor,
106 self_: *mut C_tensor,
107 );
108 pub fn atg__adaptive_avg_pool2d_out(
109 out__: *mut *mut C_tensor,
110 out_: *mut C_tensor,
111 self_: *mut C_tensor,
112 output_size_data: *const i64,
113 output_size_len: c_int,
114 );
115 pub fn atg__adaptive_avg_pool3d(
116 out__: *mut *mut C_tensor,
117 self_: *mut C_tensor,
118 output_size_data: *const i64,
119 output_size_len: c_int,
120 );
121 pub fn atg__adaptive_avg_pool3d_backward(
122 out__: *mut *mut C_tensor,
123 grad_output_: *mut C_tensor,
124 self_: *mut C_tensor,
125 );
126 pub fn atg__adaptive_avg_pool3d_backward_out(
127 out__: *mut *mut C_tensor,
128 out_: *mut C_tensor,
129 grad_output_: *mut C_tensor,
130 self_: *mut C_tensor,
131 );
132 pub fn atg__adaptive_avg_pool3d_out(
133 out__: *mut *mut C_tensor,
134 out_: *mut C_tensor,
135 self_: *mut C_tensor,
136 output_size_data: *const i64,
137 output_size_len: c_int,
138 );
139 pub fn atg__add_batch_dim(
140 out__: *mut *mut C_tensor,
141 self_: *mut C_tensor,
142 batch_dim_: i64,
143 level_: i64,
144 );
145 pub fn atg__add_relu(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
146 pub fn atg__add_relu_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
147 pub fn atg__add_relu_out(
148 out__: *mut *mut C_tensor,
149 out_: *mut C_tensor,
150 self_: *mut C_tensor,
151 other_: *mut C_tensor,
152 );
153 pub fn atg__add_relu_scalar(
154 out__: *mut *mut C_tensor,
155 self_: *mut C_tensor,
156 other_: *mut C_scalar,
157 );
158 pub fn atg__add_relu_scalar_(
159 out__: *mut *mut C_tensor,
160 self_: *mut C_tensor,
161 other_: *mut C_scalar,
162 );
163 pub fn atg__add_relu_scalar_out(
164 out__: *mut *mut C_tensor,
165 out_: *mut C_tensor,
166 self_: *mut C_tensor,
167 other_: *mut C_scalar,
168 );
169 pub fn atg__addmm_activation(
170 out__: *mut *mut C_tensor,
171 self_: *mut C_tensor,
172 mat1_: *mut C_tensor,
173 mat2_: *mut C_tensor,
174 use_gelu_: c_int,
175 );
176 pub fn atg__addmm_activation_out(
177 out__: *mut *mut C_tensor,
178 out_: *mut C_tensor,
179 self_: *mut C_tensor,
180 mat1_: *mut C_tensor,
181 mat2_: *mut C_tensor,
182 use_gelu_: c_int,
183 );
184 pub fn atg__aminmax(out__: *mut *mut C_tensor, self_: *mut C_tensor);
185 pub fn atg__aminmax_dim(
186 out__: *mut *mut C_tensor,
187 self_: *mut C_tensor,
188 dim_: i64,
189 keepdim_: c_int,
190 );
191 pub fn atg__aminmax_dim_out(
192 out__: *mut *mut C_tensor,
193 out0_: *mut C_tensor,
194 out1_: *mut C_tensor,
195 self_: *mut C_tensor,
196 dim_: i64,
197 keepdim_: c_int,
198 );
199 pub fn atg__aminmax_out(
200 out__: *mut *mut C_tensor,
201 out0_: *mut C_tensor,
202 out1_: *mut C_tensor,
203 self_: *mut C_tensor,
204 );
205 pub fn atg__amp_update_scale(
206 out__: *mut *mut C_tensor,
207 self_: *mut C_tensor,
208 growth_tracker_: *mut C_tensor,
209 found_inf_: *mut C_tensor,
210 scale_growth_factor_: f64,
211 scale_backoff_factor_: f64,
212 growth_interval_: i64,
213 );
214 pub fn atg__amp_update_scale_(
215 out__: *mut *mut C_tensor,
216 self_: *mut C_tensor,
217 growth_tracker_: *mut C_tensor,
218 found_inf_: *mut C_tensor,
219 scale_growth_factor_: f64,
220 scale_backoff_factor_: f64,
221 growth_interval_: i64,
222 );
223 pub fn atg__amp_update_scale_out(
224 out__: *mut *mut C_tensor,
225 out_: *mut C_tensor,
226 self_: *mut C_tensor,
227 growth_tracker_: *mut C_tensor,
228 found_inf_: *mut C_tensor,
229 scale_growth_factor_: f64,
230 scale_backoff_factor_: f64,
231 growth_interval_: i64,
232 );
233 pub fn atg__assert_scalar(
234 self_scalar_: *mut C_scalar,
235 assert_msg_ptr: *const u8,
236 assert_msg_len: c_int,
237 );
238 pub fn atg__assert_tensor_metadata(
239 a_: *mut C_tensor,
240 size_data: *const i64,
241 size_len: c_int,
242 stride_data: *const i64,
243 stride_len: c_int,
244 dtype_: c_int,
245 device_: c_int,
246 layout_: i8,
247 );
248 pub fn atg__autocast_to_full_precision(
249 out__: *mut *mut C_tensor,
250 self_: *mut C_tensor,
251 cuda_enabled_: c_int,
252 cpu_enabled_: c_int,
253 );
254 pub fn atg__autocast_to_reduced_precision(
255 out__: *mut *mut C_tensor,
256 self_: *mut C_tensor,
257 cuda_enabled_: c_int,
258 cpu_enabled_: c_int,
259 cuda_dtype_: c_int,
260 cpu_dtype_: c_int,
261 );
262 pub fn atg__batch_norm_no_update(
263 out__: *mut *mut C_tensor,
264 input_: *mut C_tensor,
265 weight_: *mut C_tensor,
266 bias_: *mut C_tensor,
267 running_mean_: *mut C_tensor,
268 running_var_: *mut C_tensor,
269 momentum_: f64,
270 eps_: f64,
271 );
272 pub fn atg__batch_norm_no_update_out(
273 out__: *mut *mut C_tensor,
274 out0_: *mut C_tensor,
275 out1_: *mut C_tensor,
276 out2_: *mut C_tensor,
277 out3_: *mut C_tensor,
278 input_: *mut C_tensor,
279 weight_: *mut C_tensor,
280 bias_: *mut C_tensor,
281 running_mean_: *mut C_tensor,
282 running_var_: *mut C_tensor,
283 momentum_: f64,
284 eps_: f64,
285 );
286 pub fn atg__batch_norm_with_update(
287 out__: *mut *mut C_tensor,
288 input_: *mut C_tensor,
289 weight_: *mut C_tensor,
290 bias_: *mut C_tensor,
291 running_mean_: *mut C_tensor,
292 running_var_: *mut C_tensor,
293 momentum_: f64,
294 eps_: f64,
295 );
296 pub fn atg__batch_norm_with_update_functional(
297 out__: *mut *mut C_tensor,
298 input_: *mut C_tensor,
299 weight_: *mut C_tensor,
300 bias_: *mut C_tensor,
301 running_mean_: *mut C_tensor,
302 running_var_: *mut C_tensor,
303 momentum_: f64,
304 eps_: f64,
305 );
306 pub fn atg__batch_norm_with_update_out(
307 out__: *mut *mut C_tensor,
308 out_: *mut C_tensor,
309 save_mean_: *mut C_tensor,
310 save_invstd_: *mut C_tensor,
311 reserve_: *mut C_tensor,
312 input_: *mut C_tensor,
313 weight_: *mut C_tensor,
314 bias_: *mut C_tensor,
315 running_mean_: *mut C_tensor,
316 running_var_: *mut C_tensor,
317 momentum_: f64,
318 eps_: f64,
319 );
320 pub fn atg__cast_byte(out__: *mut *mut C_tensor, self_: *mut C_tensor, non_blocking_: c_int);
321 pub fn atg__cast_char(out__: *mut *mut C_tensor, self_: *mut C_tensor, non_blocking_: c_int);
322 pub fn atg__cast_double(out__: *mut *mut C_tensor, self_: *mut C_tensor, non_blocking_: c_int);
323 pub fn atg__cast_float(out__: *mut *mut C_tensor, self_: *mut C_tensor, non_blocking_: c_int);
324 pub fn atg__cast_half(out__: *mut *mut C_tensor, self_: *mut C_tensor, non_blocking_: c_int);
325 pub fn atg__cast_int(out__: *mut *mut C_tensor, self_: *mut C_tensor, non_blocking_: c_int);
326 pub fn atg__cast_long(out__: *mut *mut C_tensor, self_: *mut C_tensor, non_blocking_: c_int);
327 pub fn atg__cast_short(out__: *mut *mut C_tensor, self_: *mut C_tensor, non_blocking_: c_int);
328 pub fn atg__cdist_backward(
329 out__: *mut *mut C_tensor,
330 grad_: *mut C_tensor,
331 x1_: *mut C_tensor,
332 x2_: *mut C_tensor,
333 p_: f64,
334 cdist_: *mut C_tensor,
335 );
336 pub fn atg__cdist_backward_out(
337 out__: *mut *mut C_tensor,
338 out_: *mut C_tensor,
339 grad_: *mut C_tensor,
340 x1_: *mut C_tensor,
341 x2_: *mut C_tensor,
342 p_: f64,
343 cdist_: *mut C_tensor,
344 );
345 pub fn atg__cholesky_solve_helper(
346 out__: *mut *mut C_tensor,
347 self_: *mut C_tensor,
348 A_: *mut C_tensor,
349 upper_: c_int,
350 );
351 pub fn atg__cholesky_solve_helper_out(
352 out__: *mut *mut C_tensor,
353 out_: *mut C_tensor,
354 self_: *mut C_tensor,
355 A_: *mut C_tensor,
356 upper_: c_int,
357 );
358 pub fn atg__chunk_cat(
359 out__: *mut *mut C_tensor,
360 tensors_data: *const *mut C_tensor,
361 tensors_len: c_int,
362 dim_: i64,
363 num_chunks_: i64,
364 );
365 pub fn atg__chunk_cat_out(
366 out__: *mut *mut C_tensor,
367 out_: *mut C_tensor,
368 tensors_data: *const *mut C_tensor,
369 tensors_len: c_int,
370 dim_: i64,
371 num_chunks_: i64,
372 );
373 pub fn atg__coalesce(out__: *mut *mut C_tensor, self_: *mut C_tensor);
374 pub fn atg__coalesce_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
375 pub fn atg__coalesced(out__: *mut *mut C_tensor, self_: *mut C_tensor, coalesced_: c_int);
376 pub fn atg__coalesced_(out__: *mut *mut C_tensor, self_: *mut C_tensor, coalesced_: c_int);
377 pub fn atg__coalesced_out(
378 out__: *mut *mut C_tensor,
379 out_: *mut C_tensor,
380 self_: *mut C_tensor,
381 coalesced_: c_int,
382 );
383 pub fn atg__compute_linear_combination(
384 out__: *mut *mut C_tensor,
385 input_: *mut C_tensor,
386 coefficients_: *mut C_tensor,
387 );
388 pub fn atg__compute_linear_combination_out(
389 out__: *mut *mut C_tensor,
390 out_: *mut C_tensor,
391 input_: *mut C_tensor,
392 coefficients_: *mut C_tensor,
393 );
394 pub fn atg__conj(out__: *mut *mut C_tensor, self_: *mut C_tensor);
395 pub fn atg__conj_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
396 pub fn atg__conj_copy_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
397 pub fn atg__conj_physical(out__: *mut *mut C_tensor, self_: *mut C_tensor);
398 pub fn atg__conj_physical_out(
399 out__: *mut *mut C_tensor,
400 out_: *mut C_tensor,
401 self_: *mut C_tensor,
402 );
403 pub fn atg__conv_depthwise2d(
404 out__: *mut *mut C_tensor,
405 self_: *mut C_tensor,
406 weight_: *mut C_tensor,
407 kernel_size_data: *const i64,
408 kernel_size_len: c_int,
409 bias_: *mut C_tensor,
410 stride_data: *const i64,
411 stride_len: c_int,
412 padding_data: *const i64,
413 padding_len: c_int,
414 dilation_data: *const i64,
415 dilation_len: c_int,
416 );
417 pub fn atg__conv_depthwise2d_out(
418 out__: *mut *mut C_tensor,
419 out_: *mut C_tensor,
420 self_: *mut C_tensor,
421 weight_: *mut C_tensor,
422 kernel_size_data: *const i64,
423 kernel_size_len: c_int,
424 bias_: *mut C_tensor,
425 stride_data: *const i64,
426 stride_len: c_int,
427 padding_data: *const i64,
428 padding_len: c_int,
429 dilation_data: *const i64,
430 dilation_len: c_int,
431 );
432 pub fn atg__convert_indices_from_coo_to_csr(
433 out__: *mut *mut C_tensor,
434 self_: *mut C_tensor,
435 size_: i64,
436 out_int32_: c_int,
437 );
438 pub fn atg__convert_indices_from_coo_to_csr_out(
439 out__: *mut *mut C_tensor,
440 out_: *mut C_tensor,
441 self_: *mut C_tensor,
442 size_: i64,
443 out_int32_: c_int,
444 );
445 pub fn atg__convert_indices_from_csr_to_coo(
446 out__: *mut *mut C_tensor,
447 crow_indices_: *mut C_tensor,
448 col_indices_: *mut C_tensor,
449 out_int32_: c_int,
450 transpose_: c_int,
451 );
452 pub fn atg__convert_indices_from_csr_to_coo_out(
453 out__: *mut *mut C_tensor,
454 out_: *mut C_tensor,
455 crow_indices_: *mut C_tensor,
456 col_indices_: *mut C_tensor,
457 out_int32_: c_int,
458 transpose_: c_int,
459 );
460 pub fn atg__convert_weight_to_int4pack(
461 out__: *mut *mut C_tensor,
462 self_: *mut C_tensor,
463 innerKTiles_: i64,
464 );
465 pub fn atg__convert_weight_to_int4pack_for_cpu(
466 out__: *mut *mut C_tensor,
467 self_: *mut C_tensor,
468 innerKTiles_: i64,
469 );
470 pub fn atg__convolution(
471 out__: *mut *mut C_tensor,
472 input_: *mut C_tensor,
473 weight_: *mut C_tensor,
474 bias_: *mut C_tensor,
475 stride_data: *const i64,
476 stride_len: c_int,
477 padding_data: *const i64,
478 padding_len: c_int,
479 dilation_data: *const i64,
480 dilation_len: c_int,
481 transposed_: c_int,
482 output_padding_data: *const i64,
483 output_padding_len: c_int,
484 groups_: i64,
485 benchmark_: c_int,
486 deterministic_: c_int,
487 cudnn_enabled_: c_int,
488 allow_tf32_: c_int,
489 );
490 pub fn atg__convolution_deprecated(
491 out__: *mut *mut C_tensor,
492 input_: *mut C_tensor,
493 weight_: *mut C_tensor,
494 bias_: *mut C_tensor,
495 stride_data: *const i64,
496 stride_len: c_int,
497 padding_data: *const i64,
498 padding_len: c_int,
499 dilation_data: *const i64,
500 dilation_len: c_int,
501 transposed_: c_int,
502 output_padding_data: *const i64,
503 output_padding_len: c_int,
504 groups_: i64,
505 benchmark_: c_int,
506 deterministic_: c_int,
507 cudnn_enabled_: c_int,
508 );
509 pub fn atg__convolution_mode(
510 out__: *mut *mut C_tensor,
511 input_: *mut C_tensor,
512 weight_: *mut C_tensor,
513 bias_: *mut C_tensor,
514 stride_data: *const i64,
515 stride_len: c_int,
516 padding_ptr: *const u8,
517 padding_len: c_int,
518 dilation_data: *const i64,
519 dilation_len: c_int,
520 groups_: i64,
521 );
522 pub fn atg__convolution_out(
523 out__: *mut *mut C_tensor,
524 out_: *mut C_tensor,
525 input_: *mut C_tensor,
526 weight_: *mut C_tensor,
527 bias_: *mut C_tensor,
528 stride_data: *const i64,
529 stride_len: c_int,
530 padding_data: *const i64,
531 padding_len: c_int,
532 dilation_data: *const i64,
533 dilation_len: c_int,
534 transposed_: c_int,
535 output_padding_data: *const i64,
536 output_padding_len: c_int,
537 groups_: i64,
538 benchmark_: c_int,
539 deterministic_: c_int,
540 cudnn_enabled_: c_int,
541 allow_tf32_: c_int,
542 );
543 pub fn atg__copy_from(
544 out__: *mut *mut C_tensor,
545 self_: *mut C_tensor,
546 dst_: *mut C_tensor,
547 non_blocking_: c_int,
548 );
549 pub fn atg__copy_from_and_resize(
550 out__: *mut *mut C_tensor,
551 self_: *mut C_tensor,
552 dst_: *mut C_tensor,
553 );
554 pub fn atg__copy_from_and_resize_out(
555 out__: *mut *mut C_tensor,
556 out_: *mut C_tensor,
557 self_: *mut C_tensor,
558 dst_: *mut C_tensor,
559 );
560 pub fn atg__copy_from_out(
561 out__: *mut *mut C_tensor,
562 out_: *mut C_tensor,
563 self_: *mut C_tensor,
564 dst_: *mut C_tensor,
565 non_blocking_: c_int,
566 );
567 pub fn atg__cslt_compress(out__: *mut *mut C_tensor, input_: *mut C_tensor);
568 pub fn atg__cslt_sparse_mm(
569 out__: *mut *mut C_tensor,
570 compressed_A_: *mut C_tensor,
571 dense_B_: *mut C_tensor,
572 bias_: *mut C_tensor,
573 alpha_: *mut C_tensor,
574 out_dtype_: c_int,
575 transpose_result_: c_int,
576 alg_id_: i64,
577 split_k_: i64,
578 split_k_mode_: i64,
579 );
580 pub fn atg__cslt_sparse_mm_search(
581 compressed_A_: *mut C_tensor,
582 dense_B_: *mut C_tensor,
583 bias_: *mut C_tensor,
584 alpha_: *mut C_tensor,
585 out_dtype_: c_int,
586 transpose_result_: c_int,
587 ) -> i64;
588 pub fn atg__ctc_loss(
589 out__: *mut *mut C_tensor,
590 log_probs_: *mut C_tensor,
591 targets_: *mut C_tensor,
592 input_lengths_data: *const i64,
593 input_lengths_len: c_int,
594 target_lengths_data: *const i64,
595 target_lengths_len: c_int,
596 blank_: i64,
597 zero_infinity_: c_int,
598 );
599 pub fn atg__ctc_loss_backward(
600 out__: *mut *mut C_tensor,
601 grad_: *mut C_tensor,
602 log_probs_: *mut C_tensor,
603 targets_: *mut C_tensor,
604 input_lengths_data: *const i64,
605 input_lengths_len: c_int,
606 target_lengths_data: *const i64,
607 target_lengths_len: c_int,
608 neg_log_likelihood_: *mut C_tensor,
609 log_alpha_: *mut C_tensor,
610 blank_: i64,
611 zero_infinity_: c_int,
612 );
613 pub fn atg__ctc_loss_backward_out(
614 out__: *mut *mut C_tensor,
615 out_: *mut C_tensor,
616 grad_: *mut C_tensor,
617 log_probs_: *mut C_tensor,
618 targets_: *mut C_tensor,
619 input_lengths_data: *const i64,
620 input_lengths_len: c_int,
621 target_lengths_data: *const i64,
622 target_lengths_len: c_int,
623 neg_log_likelihood_: *mut C_tensor,
624 log_alpha_: *mut C_tensor,
625 blank_: i64,
626 zero_infinity_: c_int,
627 );
628 pub fn atg__ctc_loss_backward_tensor(
629 out__: *mut *mut C_tensor,
630 grad_: *mut C_tensor,
631 log_probs_: *mut C_tensor,
632 targets_: *mut C_tensor,
633 input_lengths_: *mut C_tensor,
634 target_lengths_: *mut C_tensor,
635 neg_log_likelihood_: *mut C_tensor,
636 log_alpha_: *mut C_tensor,
637 blank_: i64,
638 zero_infinity_: c_int,
639 );
640 pub fn atg__ctc_loss_out(
641 out__: *mut *mut C_tensor,
642 out0_: *mut C_tensor,
643 out1_: *mut C_tensor,
644 log_probs_: *mut C_tensor,
645 targets_: *mut C_tensor,
646 input_lengths_data: *const i64,
647 input_lengths_len: c_int,
648 target_lengths_data: *const i64,
649 target_lengths_len: c_int,
650 blank_: i64,
651 zero_infinity_: c_int,
652 );
653 pub fn atg__ctc_loss_tensor(
654 out__: *mut *mut C_tensor,
655 log_probs_: *mut C_tensor,
656 targets_: *mut C_tensor,
657 input_lengths_: *mut C_tensor,
658 target_lengths_: *mut C_tensor,
659 blank_: i64,
660 zero_infinity_: c_int,
661 );
662 pub fn atg__ctc_loss_tensor_out(
663 out__: *mut *mut C_tensor,
664 out0_: *mut C_tensor,
665 out1_: *mut C_tensor,
666 log_probs_: *mut C_tensor,
667 targets_: *mut C_tensor,
668 input_lengths_: *mut C_tensor,
669 target_lengths_: *mut C_tensor,
670 blank_: i64,
671 zero_infinity_: c_int,
672 );
673 pub fn atg__cudnn_attention_backward(
674 out__: *mut *mut C_tensor,
675 grad_out_: *mut C_tensor,
676 query_: *mut C_tensor,
677 key_: *mut C_tensor,
678 value_: *mut C_tensor,
679 out_: *mut C_tensor,
680 logsumexp_: *mut C_tensor,
681 philox_seed_: *mut C_tensor,
682 philox_offset_: *mut C_tensor,
683 attn_bias_: *mut C_tensor,
684 cum_seq_q_: *mut C_tensor,
685 cum_seq_k_: *mut C_tensor,
686 max_q_: i64,
687 max_k_: i64,
688 dropout_p_: f64,
689 is_causal_: c_int,
690 scale_v: f64,
691 scale_null: i8,
692 );
693 pub fn atg__cudnn_ctc_loss(
694 out__: *mut *mut C_tensor,
695 log_probs_: *mut C_tensor,
696 targets_: *mut C_tensor,
697 input_lengths_data: *const i64,
698 input_lengths_len: c_int,
699 target_lengths_data: *const i64,
700 target_lengths_len: c_int,
701 blank_: i64,
702 deterministic_: c_int,
703 zero_infinity_: c_int,
704 );
705 pub fn atg__cudnn_ctc_loss_out(
706 out__: *mut *mut C_tensor,
707 out0_: *mut C_tensor,
708 out1_: *mut C_tensor,
709 log_probs_: *mut C_tensor,
710 targets_: *mut C_tensor,
711 input_lengths_data: *const i64,
712 input_lengths_len: c_int,
713 target_lengths_data: *const i64,
714 target_lengths_len: c_int,
715 blank_: i64,
716 deterministic_: c_int,
717 zero_infinity_: c_int,
718 );
719 pub fn atg__cudnn_ctc_loss_tensor(
720 out__: *mut *mut C_tensor,
721 log_probs_: *mut C_tensor,
722 targets_: *mut C_tensor,
723 input_lengths_: *mut C_tensor,
724 target_lengths_: *mut C_tensor,
725 blank_: i64,
726 deterministic_: c_int,
727 zero_infinity_: c_int,
728 );
729 pub fn atg__cudnn_init_dropout_state(
730 out__: *mut *mut C_tensor,
731 dropout_: f64,
732 train_: c_int,
733 dropout_seed_: i64,
734 options_kind: c_int,
735 options_device: c_int,
736 );
737 pub fn atg__cudnn_init_dropout_state_out(
738 out__: *mut *mut C_tensor,
739 out_: *mut C_tensor,
740 dropout_: f64,
741 train_: c_int,
742 dropout_seed_: i64,
743 );
744 pub fn atg__cudnn_rnn(
745 out__: *mut *mut C_tensor,
746 input_: *mut C_tensor,
747 weight_data: *const *mut C_tensor,
748 weight_len: c_int,
749 weight_stride0_: i64,
750 weight_buf_: *mut C_tensor,
751 hx_: *mut C_tensor,
752 cx_: *mut C_tensor,
753 mode_: i64,
754 hidden_size_: i64,
755 proj_size_: i64,
756 num_layers_: i64,
757 batch_first_: c_int,
758 dropout_: f64,
759 train_: c_int,
760 bidirectional_: c_int,
761 batch_sizes_data: *const i64,
762 batch_sizes_len: c_int,
763 dropout_state_: *mut C_tensor,
764 );
765 pub fn atg__cudnn_rnn_flatten_weight(
766 out__: *mut *mut C_tensor,
767 weight_arr_data: *const *mut C_tensor,
768 weight_arr_len: c_int,
769 weight_stride0_: i64,
770 input_size_: i64,
771 mode_: i64,
772 hidden_size_: i64,
773 proj_size_: i64,
774 num_layers_: i64,
775 batch_first_: c_int,
776 bidirectional_: c_int,
777 );
778 pub fn atg__cudnn_rnn_flatten_weight_out(
779 out__: *mut *mut C_tensor,
780 out_: *mut C_tensor,
781 weight_arr_data: *const *mut C_tensor,
782 weight_arr_len: c_int,
783 weight_stride0_: i64,
784 input_size_: i64,
785 mode_: i64,
786 hidden_size_: i64,
787 proj_size_: i64,
788 num_layers_: i64,
789 batch_first_: c_int,
790 bidirectional_: c_int,
791 );
792 pub fn atg__cudnn_rnn_out(
793 out__: *mut *mut C_tensor,
794 out0_: *mut C_tensor,
795 out1_: *mut C_tensor,
796 out2_: *mut C_tensor,
797 out3_: *mut C_tensor,
798 out4_: *mut C_tensor,
799 input_: *mut C_tensor,
800 weight_data: *const *mut C_tensor,
801 weight_len: c_int,
802 weight_stride0_: i64,
803 weight_buf_: *mut C_tensor,
804 hx_: *mut C_tensor,
805 cx_: *mut C_tensor,
806 mode_: i64,
807 hidden_size_: i64,
808 proj_size_: i64,
809 num_layers_: i64,
810 batch_first_: c_int,
811 dropout_: f64,
812 train_: c_int,
813 bidirectional_: c_int,
814 batch_sizes_data: *const i64,
815 batch_sizes_len: c_int,
816 dropout_state_: *mut C_tensor,
817 );
818 pub fn atg__debug_has_internal_overlap(self_: *mut C_tensor) -> i64;
819 pub fn atg__dim_arange(out__: *mut *mut C_tensor, like_: *mut C_tensor, dim_: i64);
820 pub fn atg__dimi(self_: *mut C_tensor) -> i64;
821 pub fn atg__dimv(self_: *mut C_tensor) -> i64;
822 pub fn atg__dirichlet_grad(
823 out__: *mut *mut C_tensor,
824 x_: *mut C_tensor,
825 alpha_: *mut C_tensor,
826 total_: *mut C_tensor,
827 );
828 pub fn atg__dirichlet_grad_out(
829 out__: *mut *mut C_tensor,
830 out_: *mut C_tensor,
831 x_: *mut C_tensor,
832 alpha_: *mut C_tensor,
833 total_: *mut C_tensor,
834 );
835 pub fn atg__dyn_quant_matmul_4bit(
836 out__: *mut *mut C_tensor,
837 inp_: *mut C_tensor,
838 packed_weights_: *mut C_tensor,
839 block_size_: i64,
840 in_features_: i64,
841 out_features_: i64,
842 );
843 pub fn atg__dyn_quant_pack_4bit_weight(
844 out__: *mut *mut C_tensor,
845 weights_: *mut C_tensor,
846 scales_zeros_: *mut C_tensor,
847 bias_: *mut C_tensor,
848 block_size_: i64,
849 in_features_: i64,
850 out_features_: i64,
851 );
852 pub fn atg__efficient_attention_backward(
853 out__: *mut *mut C_tensor,
854 grad_out__: *mut C_tensor,
855 query_: *mut C_tensor,
856 key_: *mut C_tensor,
857 value_: *mut C_tensor,
858 bias_: *mut C_tensor,
859 out_: *mut C_tensor,
860 cu_seqlens_q_: *mut C_tensor,
861 cu_seqlens_k_: *mut C_tensor,
862 max_seqlen_q_: i64,
863 max_seqlen_k_: i64,
864 logsumexp_: *mut C_tensor,
865 dropout_p_: f64,
866 philox_seed_: *mut C_tensor,
867 philox_offset_: *mut C_tensor,
868 custom_mask_type_: i64,
869 bias_requires_grad_: c_int,
870 scale_v: f64,
871 scale_null: i8,
872 num_splits_key_v: i64,
873 num_splits_key_null: i8,
874 window_size_v: i64,
875 window_size_null: i8,
876 shared_storage_dqdkdv_: c_int,
877 );
878 pub fn atg__efficientzerotensor(
879 out__: *mut *mut C_tensor,
880 size_data: *const i64,
881 size_len: c_int,
882 options_kind: c_int,
883 options_device: c_int,
884 );
885 pub fn atg__efficientzerotensor_out(
886 out__: *mut *mut C_tensor,
887 out_: *mut C_tensor,
888 size_data: *const i64,
889 size_len: c_int,
890 );
891 pub fn atg__embedding_bag(
892 out__: *mut *mut C_tensor,
893 weight_: *mut C_tensor,
894 indices_: *mut C_tensor,
895 offsets_: *mut C_tensor,
896 scale_grad_by_freq_: c_int,
897 mode_: i64,
898 sparse_: c_int,
899 per_sample_weights_: *mut C_tensor,
900 include_last_offset_: c_int,
901 padding_idx_: i64,
902 );
903 pub fn atg__embedding_bag_backward(
904 out__: *mut *mut C_tensor,
905 grad_: *mut C_tensor,
906 indices_: *mut C_tensor,
907 offsets_: *mut C_tensor,
908 offset2bag_: *mut C_tensor,
909 bag_size_: *mut C_tensor,
910 maximum_indices_: *mut C_tensor,
911 num_weights_: i64,
912 scale_grad_by_freq_: c_int,
913 mode_: i64,
914 sparse_: c_int,
915 per_sample_weights_: *mut C_tensor,
916 padding_idx_: i64,
917 );
918 pub fn atg__embedding_bag_dense_backward(
919 out__: *mut *mut C_tensor,
920 grad_: *mut C_tensor,
921 indices_: *mut C_tensor,
922 offset2bag_: *mut C_tensor,
923 bag_size_: *mut C_tensor,
924 maximum_indices_: *mut C_tensor,
925 num_weights_: i64,
926 scale_grad_by_freq_: c_int,
927 mode_: i64,
928 per_sample_weights_: *mut C_tensor,
929 padding_idx_: i64,
930 );
931 pub fn atg__embedding_bag_dense_backward_out(
932 out__: *mut *mut C_tensor,
933 out_: *mut C_tensor,
934 grad_: *mut C_tensor,
935 indices_: *mut C_tensor,
936 offset2bag_: *mut C_tensor,
937 bag_size_: *mut C_tensor,
938 maximum_indices_: *mut C_tensor,
939 num_weights_: i64,
940 scale_grad_by_freq_: c_int,
941 mode_: i64,
942 per_sample_weights_: *mut C_tensor,
943 padding_idx_: i64,
944 );
945 pub fn atg__embedding_bag_forward_only(
946 out__: *mut *mut C_tensor,
947 weight_: *mut C_tensor,
948 indices_: *mut C_tensor,
949 offsets_: *mut C_tensor,
950 scale_grad_by_freq_: c_int,
951 mode_: i64,
952 sparse_: c_int,
953 per_sample_weights_: *mut C_tensor,
954 include_last_offset_: c_int,
955 padding_idx_: i64,
956 );
957 pub fn atg__embedding_bag_forward_only_out(
958 out__: *mut *mut C_tensor,
959 out0_: *mut C_tensor,
960 out1_: *mut C_tensor,
961 out2_: *mut C_tensor,
962 out3_: *mut C_tensor,
963 weight_: *mut C_tensor,
964 indices_: *mut C_tensor,
965 offsets_: *mut C_tensor,
966 scale_grad_by_freq_: c_int,
967 mode_: i64,
968 sparse_: c_int,
969 per_sample_weights_: *mut C_tensor,
970 include_last_offset_: c_int,
971 padding_idx_: i64,
972 );
973 pub fn atg__embedding_bag_out(
974 out__: *mut *mut C_tensor,
975 out0_: *mut C_tensor,
976 out1_: *mut C_tensor,
977 out2_: *mut C_tensor,
978 out3_: *mut C_tensor,
979 weight_: *mut C_tensor,
980 indices_: *mut C_tensor,
981 offsets_: *mut C_tensor,
982 scale_grad_by_freq_: c_int,
983 mode_: i64,
984 sparse_: c_int,
985 per_sample_weights_: *mut C_tensor,
986 include_last_offset_: c_int,
987 padding_idx_: i64,
988 );
989 pub fn atg__embedding_bag_per_sample_weights_backward(
990 out__: *mut *mut C_tensor,
991 grad_: *mut C_tensor,
992 weight_: *mut C_tensor,
993 indices_: *mut C_tensor,
994 offsets_: *mut C_tensor,
995 offset2bag_: *mut C_tensor,
996 mode_: i64,
997 padding_idx_: i64,
998 );
999 pub fn atg__embedding_bag_per_sample_weights_backward_out(
1000 out__: *mut *mut C_tensor,
1001 out_: *mut C_tensor,
1002 grad_: *mut C_tensor,
1003 weight_: *mut C_tensor,
1004 indices_: *mut C_tensor,
1005 offsets_: *mut C_tensor,
1006 offset2bag_: *mut C_tensor,
1007 mode_: i64,
1008 padding_idx_: i64,
1009 );
1010 pub fn atg__embedding_bag_sparse_backward(
1011 out__: *mut *mut C_tensor,
1012 grad_: *mut C_tensor,
1013 indices_: *mut C_tensor,
1014 offsets_: *mut C_tensor,
1015 offset2bag_: *mut C_tensor,
1016 bag_size_: *mut C_tensor,
1017 num_weights_: i64,
1018 scale_grad_by_freq_: c_int,
1019 mode_: i64,
1020 per_sample_weights_: *mut C_tensor,
1021 padding_idx_: i64,
1022 );
1023 pub fn atg__empty_affine_quantized(
1024 out__: *mut *mut C_tensor,
1025 size_data: *const i64,
1026 size_len: c_int,
1027 options_kind: c_int,
1028 options_device: c_int,
1029 scale_: f64,
1030 zero_point_: i64,
1031 );
1032 pub fn atg__empty_affine_quantized_out(
1033 out__: *mut *mut C_tensor,
1034 out_: *mut C_tensor,
1035 size_data: *const i64,
1036 size_len: c_int,
1037 scale_: f64,
1038 zero_point_: i64,
1039 );
1040 pub fn atg__empty_per_channel_affine_quantized(
1041 out__: *mut *mut C_tensor,
1042 size_data: *const i64,
1043 size_len: c_int,
1044 scales_: *mut C_tensor,
1045 zero_points_: *mut C_tensor,
1046 axis_: i64,
1047 options_kind: c_int,
1048 options_device: c_int,
1049 );
1050 pub fn atg__empty_per_channel_affine_quantized_out(
1051 out__: *mut *mut C_tensor,
1052 out_: *mut C_tensor,
1053 size_data: *const i64,
1054 size_len: c_int,
1055 scales_: *mut C_tensor,
1056 zero_points_: *mut C_tensor,
1057 axis_: i64,
1058 );
1059 pub fn atg__euclidean_dist(out__: *mut *mut C_tensor, x1_: *mut C_tensor, x2_: *mut C_tensor);
1060 pub fn atg__euclidean_dist_out(
1061 out__: *mut *mut C_tensor,
1062 out_: *mut C_tensor,
1063 x1_: *mut C_tensor,
1064 x2_: *mut C_tensor,
1065 );
1066 pub fn atg__fake_quantize_learnable_per_channel_affine(
1067 out__: *mut *mut C_tensor,
1068 self_: *mut C_tensor,
1069 scale_: *mut C_tensor,
1070 zero_point_: *mut C_tensor,
1071 axis_: i64,
1072 quant_min_: i64,
1073 quant_max_: i64,
1074 grad_factor_: f64,
1075 );
1076 pub fn atg__fake_quantize_learnable_per_channel_affine_backward(
1077 out__: *mut *mut C_tensor,
1078 grad_: *mut C_tensor,
1079 self_: *mut C_tensor,
1080 scale_: *mut C_tensor,
1081 zero_point_: *mut C_tensor,
1082 axis_: i64,
1083 quant_min_: i64,
1084 quant_max_: i64,
1085 grad_factor_: f64,
1086 );
1087 pub fn atg__fake_quantize_learnable_per_channel_affine_out(
1088 out__: *mut *mut C_tensor,
1089 out_: *mut C_tensor,
1090 self_: *mut C_tensor,
1091 scale_: *mut C_tensor,
1092 zero_point_: *mut C_tensor,
1093 axis_: i64,
1094 quant_min_: i64,
1095 quant_max_: i64,
1096 grad_factor_: f64,
1097 );
1098 pub fn atg__fake_quantize_learnable_per_tensor_affine(
1099 out__: *mut *mut C_tensor,
1100 self_: *mut C_tensor,
1101 scale_: *mut C_tensor,
1102 zero_point_: *mut C_tensor,
1103 quant_min_: i64,
1104 quant_max_: i64,
1105 grad_factor_: f64,
1106 );
1107 pub fn atg__fake_quantize_learnable_per_tensor_affine_backward(
1108 out__: *mut *mut C_tensor,
1109 grad_: *mut C_tensor,
1110 self_: *mut C_tensor,
1111 scale_: *mut C_tensor,
1112 zero_point_: *mut C_tensor,
1113 quant_min_: i64,
1114 quant_max_: i64,
1115 grad_factor_: f64,
1116 );
1117 pub fn atg__fake_quantize_learnable_per_tensor_affine_out(
1118 out__: *mut *mut C_tensor,
1119 out_: *mut C_tensor,
1120 self_: *mut C_tensor,
1121 scale_: *mut C_tensor,
1122 zero_point_: *mut C_tensor,
1123 quant_min_: i64,
1124 quant_max_: i64,
1125 grad_factor_: f64,
1126 );
1127 pub fn atg__fake_quantize_per_tensor_affine_cachemask_tensor_qparams(
1128 out__: *mut *mut C_tensor,
1129 self_: *mut C_tensor,
1130 scale_: *mut C_tensor,
1131 zero_point_: *mut C_tensor,
1132 fake_quant_enabled_: *mut C_tensor,
1133 quant_min_: i64,
1134 quant_max_: i64,
1135 );
1136 pub fn atg__fake_quantize_per_tensor_affine_cachemask_tensor_qparams_out(
1137 out__: *mut *mut C_tensor,
1138 out0_: *mut C_tensor,
1139 out1_: *mut C_tensor,
1140 self_: *mut C_tensor,
1141 scale_: *mut C_tensor,
1142 zero_point_: *mut C_tensor,
1143 fake_quant_enabled_: *mut C_tensor,
1144 quant_min_: i64,
1145 quant_max_: i64,
1146 );
1147 pub fn atg__fft_c2c(
1148 out__: *mut *mut C_tensor,
1149 self_: *mut C_tensor,
1150 dim_data: *const i64,
1151 dim_len: c_int,
1152 normalization_: i64,
1153 forward_: c_int,
1154 );
1155 pub fn atg__fft_c2c_out(
1156 out__: *mut *mut C_tensor,
1157 out_: *mut C_tensor,
1158 self_: *mut C_tensor,
1159 dim_data: *const i64,
1160 dim_len: c_int,
1161 normalization_: i64,
1162 forward_: c_int,
1163 );
1164 pub fn atg__fft_c2r(
1165 out__: *mut *mut C_tensor,
1166 self_: *mut C_tensor,
1167 dim_data: *const i64,
1168 dim_len: c_int,
1169 normalization_: i64,
1170 last_dim_size_: i64,
1171 );
1172 pub fn atg__fft_c2r_out(
1173 out__: *mut *mut C_tensor,
1174 out_: *mut C_tensor,
1175 self_: *mut C_tensor,
1176 dim_data: *const i64,
1177 dim_len: c_int,
1178 normalization_: i64,
1179 last_dim_size_: i64,
1180 );
1181 pub fn atg__fft_r2c(
1182 out__: *mut *mut C_tensor,
1183 self_: *mut C_tensor,
1184 dim_data: *const i64,
1185 dim_len: c_int,
1186 normalization_: i64,
1187 onesided_: c_int,
1188 );
1189 pub fn atg__fft_r2c_out(
1190 out__: *mut *mut C_tensor,
1191 out_: *mut C_tensor,
1192 self_: *mut C_tensor,
1193 dim_data: *const i64,
1194 dim_len: c_int,
1195 normalization_: i64,
1196 onesided_: c_int,
1197 );
1198 pub fn atg__fill_mem_eff_dropout_mask_(
1199 out__: *mut *mut C_tensor,
1200 self_: *mut C_tensor,
1201 dropout_p_: f64,
1202 seed_: i64,
1203 offset_: i64,
1204 );
1205 pub fn atg__flash_attention_backward(
1206 out__: *mut *mut C_tensor,
1207 grad_out_: *mut C_tensor,
1208 query_: *mut C_tensor,
1209 key_: *mut C_tensor,
1210 value_: *mut C_tensor,
1211 out_: *mut C_tensor,
1212 logsumexp_: *mut C_tensor,
1213 cum_seq_q_: *mut C_tensor,
1214 cum_seq_k_: *mut C_tensor,
1215 max_q_: i64,
1216 max_k_: i64,
1217 dropout_p_: f64,
1218 is_causal_: c_int,
1219 rng_state_: *mut C_tensor,
1220 unused_: *mut C_tensor,
1221 scale_v: f64,
1222 scale_null: i8,
1223 window_size_left_v: i64,
1224 window_size_left_null: i8,
1225 window_size_right_v: i64,
1226 window_size_right_null: i8,
1227 );
1228 pub fn atg__flash_attention_forward_no_dropout_inplace(
1229 out__: *mut *mut C_tensor,
1230 out_: *mut C_tensor,
1231 query_: *mut C_tensor,
1232 key_: *mut C_tensor,
1233 value_: *mut C_tensor,
1234 cum_seq_q_: *mut C_tensor,
1235 cum_seq_k_: *mut C_tensor,
1236 max_q_: i64,
1237 max_k_: i64,
1238 dropout_p_: f64,
1239 is_causal_: c_int,
1240 return_debug_mask_: c_int,
1241 scale_v: f64,
1242 scale_null: i8,
1243 window_size_left_v: i64,
1244 window_size_left_null: i8,
1245 window_size_right_v: i64,
1246 window_size_right_null: i8,
1247 seqused_k_: *mut C_tensor,
1248 alibi_slopes_: *mut C_tensor,
1249 block_table_: *mut C_tensor,
1250 num_splits_v: i64,
1251 num_splits_null: i8,
1252 );
1253 pub fn atg__foobar(
1254 out__: *mut *mut C_tensor,
1255 self_: *mut C_tensor,
1256 arg1_: c_int,
1257 arg2_: c_int,
1258 arg3_: c_int,
1259 );
1260 pub fn atg__foobar_out(
1261 out__: *mut *mut C_tensor,
1262 out_: *mut C_tensor,
1263 self_: *mut C_tensor,
1264 arg1_: c_int,
1265 arg2_: c_int,
1266 arg3_: c_int,
1267 );
1268 pub fn atg__functional_assert_async(
1269 out__: *mut *mut C_tensor,
1270 self_: *mut C_tensor,
1271 assert_msg_ptr: *const u8,
1272 assert_msg_len: c_int,
1273 dep_token_: *mut C_tensor,
1274 );
1275 pub fn atg__functional_assert_scalar(
1276 out__: *mut *mut C_tensor,
1277 self_scalar_: *mut C_scalar,
1278 assert_msg_ptr: *const u8,
1279 assert_msg_len: c_int,
1280 dep_token_: *mut C_tensor,
1281 );
1282 pub fn atg__functional_sym_constrain_range(
1283 out__: *mut *mut C_tensor,
1284 size_: *mut C_scalar,
1285 min_v: i64,
1286 min_null: i8,
1287 max_v: i64,
1288 max_null: i8,
1289 dep_token_: *mut C_tensor,
1290 );
1291 pub fn atg__functional_sym_constrain_range_for_size(
1292 out__: *mut *mut C_tensor,
1293 size_: *mut C_scalar,
1294 min_v: i64,
1295 min_null: i8,
1296 max_v: i64,
1297 max_null: i8,
1298 dep_token_: *mut C_tensor,
1299 );
1300 pub fn atg__fused_dropout(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: f64);
1301 pub fn atg__fused_dropout_out(
1302 out__: *mut *mut C_tensor,
1303 out0_: *mut C_tensor,
1304 out1_: *mut C_tensor,
1305 self_: *mut C_tensor,
1306 p_: f64,
1307 );
1308 pub fn atg__fused_moving_avg_obs_fq_helper(
1309 out__: *mut *mut C_tensor,
1310 self_: *mut C_tensor,
1311 observer_on_: *mut C_tensor,
1312 fake_quant_on_: *mut C_tensor,
1313 running_min_: *mut C_tensor,
1314 running_max_: *mut C_tensor,
1315 scale_: *mut C_tensor,
1316 zero_point_: *mut C_tensor,
1317 averaging_const_: f64,
1318 quant_min_: i64,
1319 quant_max_: i64,
1320 ch_axis_: i64,
1321 per_row_fake_quant_: c_int,
1322 symmetric_quant_: c_int,
1323 );
1324 pub fn atg__fused_moving_avg_obs_fq_helper_functional(
1325 out__: *mut *mut C_tensor,
1326 self_: *mut C_tensor,
1327 observer_on_: *mut C_tensor,
1328 fake_quant_on_: *mut C_tensor,
1329 running_min_: *mut C_tensor,
1330 running_max_: *mut C_tensor,
1331 scale_: *mut C_tensor,
1332 zero_point_: *mut C_tensor,
1333 averaging_const_: f64,
1334 quant_min_: i64,
1335 quant_max_: i64,
1336 ch_axis_: i64,
1337 per_row_fake_quant_: c_int,
1338 symmetric_quant_: c_int,
1339 );
1340 pub fn atg__fused_moving_avg_obs_fq_helper_out(
1341 out__: *mut *mut C_tensor,
1342 out0_: *mut C_tensor,
1343 out1_: *mut C_tensor,
1344 self_: *mut C_tensor,
1345 observer_on_: *mut C_tensor,
1346 fake_quant_on_: *mut C_tensor,
1347 running_min_: *mut C_tensor,
1348 running_max_: *mut C_tensor,
1349 scale_: *mut C_tensor,
1350 zero_point_: *mut C_tensor,
1351 averaging_const_: f64,
1352 quant_min_: i64,
1353 quant_max_: i64,
1354 ch_axis_: i64,
1355 per_row_fake_quant_: c_int,
1356 symmetric_quant_: c_int,
1357 );
1358 pub fn atg__fused_rms_norm(
1359 out__: *mut *mut C_tensor,
1360 input_: *mut C_tensor,
1361 normalized_shape_data: *const i64,
1362 normalized_shape_len: c_int,
1363 weight_: *mut C_tensor,
1364 eps_v: f64,
1365 eps_null: i8,
1366 );
1367 pub fn atg__fused_sdp_choice(
1368 query_: *mut C_tensor,
1369 key_: *mut C_tensor,
1370 value_: *mut C_tensor,
1371 attn_mask_: *mut C_tensor,
1372 dropout_p_: f64,
1373 is_causal_: c_int,
1374 scale_v: f64,
1375 scale_null: i8,
1376 enable_gqa_: c_int,
1377 ) -> i64;
1378 pub fn atg__fw_primal(out__: *mut *mut C_tensor, self_: *mut C_tensor, level_: i64);
1379 pub fn atg__fw_primal_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor, level_: i64);
1380 pub fn atg__fw_primal_copy_out(
1381 out__: *mut *mut C_tensor,
1382 out_: *mut C_tensor,
1383 self_: *mut C_tensor,
1384 level_: i64,
1385 );
1386 pub fn atg__gather_sparse_backward(
1387 out__: *mut *mut C_tensor,
1388 self_: *mut C_tensor,
1389 dim_: i64,
1390 index_: *mut C_tensor,
1391 grad_: *mut C_tensor,
1392 );
1393 pub fn atg__grid_sampler_2d_cpu_fallback(
1394 out__: *mut *mut C_tensor,
1395 input_: *mut C_tensor,
1396 grid_: *mut C_tensor,
1397 interpolation_mode_: i64,
1398 padding_mode_: i64,
1399 align_corners_: c_int,
1400 );
1401 pub fn atg__grid_sampler_2d_cpu_fallback_backward(
1402 out__: *mut *mut C_tensor,
1403 grad_output_: *mut C_tensor,
1404 input_: *mut C_tensor,
1405 grid_: *mut C_tensor,
1406 interpolation_mode_: i64,
1407 padding_mode_: i64,
1408 align_corners_: c_int,
1409 );
1410 pub fn atg__grid_sampler_2d_cpu_fallback_out(
1411 out__: *mut *mut C_tensor,
1412 out_: *mut C_tensor,
1413 input_: *mut C_tensor,
1414 grid_: *mut C_tensor,
1415 interpolation_mode_: i64,
1416 padding_mode_: i64,
1417 align_corners_: c_int,
1418 );
1419 pub fn atg__grouped_mm(
1420 out__: *mut *mut C_tensor,
1421 self_: *mut C_tensor,
1422 mat2_: *mut C_tensor,
1423 offs_: *mut C_tensor,
1424 bias_: *mut C_tensor,
1425 out_dtype_: c_int,
1426 );
1427 pub fn atg__has_compatible_shallow_copy_type(
1428 self_: *mut C_tensor,
1429 from_: *mut C_tensor,
1430 ) -> c_int;
1431 pub fn atg__has_same_storage_numel(self_: *mut C_tensor, other_: *mut C_tensor) -> c_int;
1432 pub fn atg__histogramdd_bin_edges(
1433 self_: *mut C_tensor,
1434 bins_data: *const i64,
1435 bins_len: c_int,
1436 range_data: *const f64,
1437 range_len: c_int,
1438 weight_: *mut C_tensor,
1439 density_: c_int,
1440 ) -> *mut *mut C_tensor;
1441 pub fn atg__histogramdd_bin_edges_out(
1442 out_data: *const *mut C_tensor,
1443 out_len: c_int,
1444 self_: *mut C_tensor,
1445 bins_data: *const i64,
1446 bins_len: c_int,
1447 range_data: *const f64,
1448 range_len: c_int,
1449 weight_: *mut C_tensor,
1450 density_: c_int,
1451 );
1452 pub fn atg__histogramdd_from_bin_cts(
1453 out__: *mut *mut C_tensor,
1454 self_: *mut C_tensor,
1455 bins_data: *const i64,
1456 bins_len: c_int,
1457 range_data: *const f64,
1458 range_len: c_int,
1459 weight_: *mut C_tensor,
1460 density_: c_int,
1461 );
1462 pub fn atg__histogramdd_from_bin_cts_out(
1463 out__: *mut *mut C_tensor,
1464 out_: *mut C_tensor,
1465 self_: *mut C_tensor,
1466 bins_data: *const i64,
1467 bins_len: c_int,
1468 range_data: *const f64,
1469 range_len: c_int,
1470 weight_: *mut C_tensor,
1471 density_: c_int,
1472 );
1473 pub fn atg__histogramdd_from_bin_tensors(
1474 out__: *mut *mut C_tensor,
1475 self_: *mut C_tensor,
1476 bins_data: *const *mut C_tensor,
1477 bins_len: c_int,
1478 weight_: *mut C_tensor,
1479 density_: c_int,
1480 );
1481 pub fn atg__histogramdd_from_bin_tensors_out(
1482 out__: *mut *mut C_tensor,
1483 out_: *mut C_tensor,
1484 self_: *mut C_tensor,
1485 bins_data: *const *mut C_tensor,
1486 bins_len: c_int,
1487 weight_: *mut C_tensor,
1488 density_: c_int,
1489 );
1490 pub fn atg__index_put_impl(
1491 out__: *mut *mut C_tensor,
1492 self_: *mut C_tensor,
1493 indices_data: *const *mut C_tensor,
1494 indices_len: c_int,
1495 values_: *mut C_tensor,
1496 accumulate_: c_int,
1497 unsafe_: c_int,
1498 );
1499 pub fn atg__index_put_impl_(
1500 out__: *mut *mut C_tensor,
1501 self_: *mut C_tensor,
1502 indices_data: *const *mut C_tensor,
1503 indices_len: c_int,
1504 values_: *mut C_tensor,
1505 accumulate_: c_int,
1506 unsafe_: c_int,
1507 );
1508 pub fn atg__index_put_impl_out(
1509 out__: *mut *mut C_tensor,
1510 out_: *mut C_tensor,
1511 self_: *mut C_tensor,
1512 indices_data: *const *mut C_tensor,
1513 indices_len: c_int,
1514 values_: *mut C_tensor,
1515 accumulate_: c_int,
1516 unsafe_: c_int,
1517 );
1518 pub fn atg__indices(out__: *mut *mut C_tensor, self_: *mut C_tensor);
1519 pub fn atg__indices_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
1520 pub fn atg__indices_copy_out(
1521 out__: *mut *mut C_tensor,
1522 out_: *mut C_tensor,
1523 self_: *mut C_tensor,
1524 );
1525 pub fn atg__int_mm(out__: *mut *mut C_tensor, self_: *mut C_tensor, mat2_: *mut C_tensor);
1526 pub fn atg__int_mm_out(
1527 out__: *mut *mut C_tensor,
1528 out_: *mut C_tensor,
1529 self_: *mut C_tensor,
1530 mat2_: *mut C_tensor,
1531 );
1532 pub fn atg__is_all_true(out__: *mut *mut C_tensor, self_: *mut C_tensor);
1533 pub fn atg__is_any_true(out__: *mut *mut C_tensor, self_: *mut C_tensor);
1534 pub fn atg__is_zerotensor(self_: *mut C_tensor) -> c_int;
1535 pub fn atg__lazy_clone(out__: *mut *mut C_tensor, self_: *mut C_tensor);
1536 pub fn atg__linalg_check_errors(
1537 info_: *mut C_tensor,
1538 api_name_ptr: *const u8,
1539 api_name_len: c_int,
1540 is_matrix_: c_int,
1541 );
1542 pub fn atg__linalg_det(out__: *mut *mut C_tensor, A_: *mut C_tensor);
1543 pub fn atg__linalg_det_result(
1544 out__: *mut *mut C_tensor,
1545 result_: *mut C_tensor,
1546 LU_: *mut C_tensor,
1547 pivots_: *mut C_tensor,
1548 A_: *mut C_tensor,
1549 );
1550 pub fn atg__linalg_eigh(
1551 out__: *mut *mut C_tensor,
1552 A_: *mut C_tensor,
1553 UPLO_ptr: *const u8,
1554 UPLO_len: c_int,
1555 compute_v_: c_int,
1556 );
1557 pub fn atg__linalg_eigh_eigenvalues(
1558 out__: *mut *mut C_tensor,
1559 eigenvalues_: *mut C_tensor,
1560 eigenvectors_: *mut C_tensor,
1561 A_: *mut C_tensor,
1562 UPLO_ptr: *const u8,
1563 UPLO_len: c_int,
1564 compute_v_: c_int,
1565 );
1566 pub fn atg__linalg_eigvals(out__: *mut *mut C_tensor, self_: *mut C_tensor);
1567 pub fn atg__linalg_slogdet(out__: *mut *mut C_tensor, A_: *mut C_tensor);
1568 pub fn atg__linalg_slogdet_sign(
1569 out__: *mut *mut C_tensor,
1570 sign_: *mut C_tensor,
1571 logabsdet_: *mut C_tensor,
1572 LU_: *mut C_tensor,
1573 pivots_: *mut C_tensor,
1574 A_: *mut C_tensor,
1575 );
1576 pub fn atg__linalg_solve_ex(
1577 out__: *mut *mut C_tensor,
1578 A_: *mut C_tensor,
1579 B_: *mut C_tensor,
1580 left_: c_int,
1581 check_errors_: c_int,
1582 );
1583 pub fn atg__linalg_solve_ex_result(
1584 out__: *mut *mut C_tensor,
1585 result_: *mut C_tensor,
1586 LU_: *mut C_tensor,
1587 pivots_: *mut C_tensor,
1588 info_: *mut C_tensor,
1589 A_: *mut C_tensor,
1590 B_: *mut C_tensor,
1591 left_: c_int,
1592 check_errors_: c_int,
1593 );
1594 pub fn atg__linalg_svd(
1595 out__: *mut *mut C_tensor,
1596 A_: *mut C_tensor,
1597 full_matrices_: c_int,
1598 compute_uv_: c_int,
1599 driver_ptr: *const u8,
1600 driver_len: c_int,
1601 );
1602 pub fn atg__linalg_svd_u(
1603 out__: *mut *mut C_tensor,
1604 U_: *mut C_tensor,
1605 S_: *mut C_tensor,
1606 Vh_: *mut C_tensor,
1607 A_: *mut C_tensor,
1608 full_matrices_: c_int,
1609 compute_uv_: c_int,
1610 driver_ptr: *const u8,
1611 driver_len: c_int,
1612 );
1613 pub fn atg__log_softmax(
1614 out__: *mut *mut C_tensor,
1615 self_: *mut C_tensor,
1616 dim_: i64,
1617 half_to_float_: c_int,
1618 );
1619 pub fn atg__log_softmax_backward_data(
1620 out__: *mut *mut C_tensor,
1621 grad_output_: *mut C_tensor,
1622 output_: *mut C_tensor,
1623 dim_: i64,
1624 input_dtype_: c_int,
1625 );
1626 pub fn atg__log_softmax_backward_data_out(
1627 out__: *mut *mut C_tensor,
1628 out_: *mut C_tensor,
1629 grad_output_: *mut C_tensor,
1630 output_: *mut C_tensor,
1631 dim_: i64,
1632 input_dtype_: c_int,
1633 );
1634 pub fn atg__log_softmax_out(
1635 out__: *mut *mut C_tensor,
1636 out_: *mut C_tensor,
1637 self_: *mut C_tensor,
1638 dim_: i64,
1639 half_to_float_: c_int,
1640 );
1641 pub fn atg__logcumsumexp(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
1642 pub fn atg__logcumsumexp_out(
1643 out__: *mut *mut C_tensor,
1644 out_: *mut C_tensor,
1645 self_: *mut C_tensor,
1646 dim_: i64,
1647 );
1648 pub fn atg__lstm_mps(
1649 out__: *mut *mut C_tensor,
1650 input_: *mut C_tensor,
1651 hx_data: *const *mut C_tensor,
1652 hx_len: c_int,
1653 params_data: *const *mut C_tensor,
1654 params_len: c_int,
1655 has_biases_: c_int,
1656 num_layers_: i64,
1657 dropout_: f64,
1658 train_: c_int,
1659 bidirectional_: c_int,
1660 batch_first_: c_int,
1661 );
1662 pub fn atg__lstm_mps_out(
1663 out__: *mut *mut C_tensor,
1664 out0_: *mut C_tensor,
1665 out1_: *mut C_tensor,
1666 out2_: *mut C_tensor,
1667 out3_: *mut C_tensor,
1668 out4_: *mut C_tensor,
1669 out5_: *mut C_tensor,
1670 input_: *mut C_tensor,
1671 hx_data: *const *mut C_tensor,
1672 hx_len: c_int,
1673 params_data: *const *mut C_tensor,
1674 params_len: c_int,
1675 has_biases_: c_int,
1676 num_layers_: i64,
1677 dropout_: f64,
1678 train_: c_int,
1679 bidirectional_: c_int,
1680 batch_first_: c_int,
1681 );
1682 pub fn atg__lu_with_info(
1683 out__: *mut *mut C_tensor,
1684 self_: *mut C_tensor,
1685 pivot_: c_int,
1686 check_errors_: c_int,
1687 );
1688 pub fn atg__make_dep_token(
1689 out__: *mut *mut C_tensor,
1690 options_kind: c_int,
1691 options_device: c_int,
1692 );
1693 pub fn atg__make_dual(
1694 out__: *mut *mut C_tensor,
1695 primal_: *mut C_tensor,
1696 tangent_: *mut C_tensor,
1697 level_: i64,
1698 );
1699 pub fn atg__make_dual_copy(
1700 out__: *mut *mut C_tensor,
1701 primal_: *mut C_tensor,
1702 tangent_: *mut C_tensor,
1703 level_: i64,
1704 );
1705 pub fn atg__make_dual_copy_out(
1706 out__: *mut *mut C_tensor,
1707 out_: *mut C_tensor,
1708 primal_: *mut C_tensor,
1709 tangent_: *mut C_tensor,
1710 level_: i64,
1711 );
1712 pub fn atg__make_per_channel_quantized_tensor(
1713 out__: *mut *mut C_tensor,
1714 self_: *mut C_tensor,
1715 scale_: *mut C_tensor,
1716 zero_point_: *mut C_tensor,
1717 axis_: i64,
1718 );
1719 pub fn atg__make_per_channel_quantized_tensor_out(
1720 out__: *mut *mut C_tensor,
1721 out_: *mut C_tensor,
1722 self_: *mut C_tensor,
1723 scale_: *mut C_tensor,
1724 zero_point_: *mut C_tensor,
1725 axis_: i64,
1726 );
1727 pub fn atg__make_per_tensor_quantized_tensor(
1728 out__: *mut *mut C_tensor,
1729 self_: *mut C_tensor,
1730 scale_: f64,
1731 zero_point_: i64,
1732 );
1733 pub fn atg__make_per_tensor_quantized_tensor_out(
1734 out__: *mut *mut C_tensor,
1735 out_: *mut C_tensor,
1736 self_: *mut C_tensor,
1737 scale_: f64,
1738 zero_point_: i64,
1739 );
1740 pub fn atg__masked_scale(
1741 out__: *mut *mut C_tensor,
1742 self_: *mut C_tensor,
1743 mask_: *mut C_tensor,
1744 scale_: f64,
1745 );
1746 pub fn atg__masked_scale_out(
1747 out__: *mut *mut C_tensor,
1748 out_: *mut C_tensor,
1749 self_: *mut C_tensor,
1750 mask_: *mut C_tensor,
1751 scale_: f64,
1752 );
1753 pub fn atg__masked_softmax(
1754 out__: *mut *mut C_tensor,
1755 self_: *mut C_tensor,
1756 mask_: *mut C_tensor,
1757 dim_v: i64,
1758 dim_null: i8,
1759 mask_type_v: i64,
1760 mask_type_null: i8,
1761 );
1762 pub fn atg__masked_softmax_backward(
1763 out__: *mut *mut C_tensor,
1764 grad_output_: *mut C_tensor,
1765 output_: *mut C_tensor,
1766 mask_: *mut C_tensor,
1767 dim_v: i64,
1768 dim_null: i8,
1769 );
1770 pub fn atg__masked_softmax_backward_out(
1771 out__: *mut *mut C_tensor,
1772 out_: *mut C_tensor,
1773 grad_output_: *mut C_tensor,
1774 output_: *mut C_tensor,
1775 mask_: *mut C_tensor,
1776 dim_v: i64,
1777 dim_null: i8,
1778 );
1779 pub fn atg__masked_softmax_out(
1780 out__: *mut *mut C_tensor,
1781 out_: *mut C_tensor,
1782 self_: *mut C_tensor,
1783 mask_: *mut C_tensor,
1784 dim_v: i64,
1785 dim_null: i8,
1786 mask_type_v: i64,
1787 mask_type_null: i8,
1788 );
1789 pub fn atg__mixed_dtypes_linear(
1790 out__: *mut *mut C_tensor,
1791 input_: *mut C_tensor,
1792 weight_: *mut C_tensor,
1793 scale_: *mut C_tensor,
1794 bias_: *mut C_tensor,
1795 activation_ptr: *const u8,
1796 activation_len: c_int,
1797 );
1798 pub fn atg__mkldnn_reshape(
1799 out__: *mut *mut C_tensor,
1800 self_: *mut C_tensor,
1801 shape_data: *const i64,
1802 shape_len: c_int,
1803 );
1804 pub fn atg__mkldnn_reshape_out(
1805 out__: *mut *mut C_tensor,
1806 out_: *mut C_tensor,
1807 self_: *mut C_tensor,
1808 shape_data: *const i64,
1809 shape_len: c_int,
1810 );
1811 pub fn atg__mkldnn_transpose(
1812 out__: *mut *mut C_tensor,
1813 self_: *mut C_tensor,
1814 dim0_: i64,
1815 dim1_: i64,
1816 );
1817 pub fn atg__mkldnn_transpose_(
1818 out__: *mut *mut C_tensor,
1819 self_: *mut C_tensor,
1820 dim0_: i64,
1821 dim1_: i64,
1822 );
1823 pub fn atg__mkldnn_transpose_out(
1824 out__: *mut *mut C_tensor,
1825 out_: *mut C_tensor,
1826 self_: *mut C_tensor,
1827 dim0_: i64,
1828 dim1_: i64,
1829 );
1830 pub fn atg__mps_convolution(
1831 out__: *mut *mut C_tensor,
1832 self_: *mut C_tensor,
1833 weight_: *mut C_tensor,
1834 bias_: *mut C_tensor,
1835 padding_data: *const i64,
1836 padding_len: c_int,
1837 stride_data: *const i64,
1838 stride_len: c_int,
1839 dilation_data: *const i64,
1840 dilation_len: c_int,
1841 groups_: i64,
1842 );
1843 pub fn atg__mps_convolution_out(
1844 out__: *mut *mut C_tensor,
1845 out_: *mut C_tensor,
1846 self_: *mut C_tensor,
1847 weight_: *mut C_tensor,
1848 bias_: *mut C_tensor,
1849 padding_data: *const i64,
1850 padding_len: c_int,
1851 stride_data: *const i64,
1852 stride_len: c_int,
1853 dilation_data: *const i64,
1854 dilation_len: c_int,
1855 groups_: i64,
1856 );
1857 pub fn atg__mps_convolution_transpose(
1858 out__: *mut *mut C_tensor,
1859 self_: *mut C_tensor,
1860 weight_: *mut C_tensor,
1861 padding_data: *const i64,
1862 padding_len: c_int,
1863 output_padding_data: *const i64,
1864 output_padding_len: c_int,
1865 stride_data: *const i64,
1866 stride_len: c_int,
1867 dilation_data: *const i64,
1868 dilation_len: c_int,
1869 groups_: i64,
1870 );
1871 pub fn atg__mps_convolution_transpose_out(
1872 out__: *mut *mut C_tensor,
1873 out_: *mut C_tensor,
1874 self_: *mut C_tensor,
1875 weight_: *mut C_tensor,
1876 padding_data: *const i64,
1877 padding_len: c_int,
1878 output_padding_data: *const i64,
1879 output_padding_len: c_int,
1880 stride_data: *const i64,
1881 stride_len: c_int,
1882 dilation_data: *const i64,
1883 dilation_len: c_int,
1884 groups_: i64,
1885 );
1886 pub fn atg__native_batch_norm_legit(
1887 out__: *mut *mut C_tensor,
1888 input_: *mut C_tensor,
1889 weight_: *mut C_tensor,
1890 bias_: *mut C_tensor,
1891 running_mean_: *mut C_tensor,
1892 running_var_: *mut C_tensor,
1893 training_: c_int,
1894 momentum_: f64,
1895 eps_: f64,
1896 );
1897 pub fn atg__native_batch_norm_legit_functional(
1898 out__: *mut *mut C_tensor,
1899 input_: *mut C_tensor,
1900 weight_: *mut C_tensor,
1901 bias_: *mut C_tensor,
1902 running_mean_: *mut C_tensor,
1903 running_var_: *mut C_tensor,
1904 training_: c_int,
1905 momentum_: f64,
1906 eps_: f64,
1907 );
1908 pub fn atg__native_batch_norm_legit_no_stats(
1909 out__: *mut *mut C_tensor,
1910 input_: *mut C_tensor,
1911 weight_: *mut C_tensor,
1912 bias_: *mut C_tensor,
1913 training_: c_int,
1914 momentum_: f64,
1915 eps_: f64,
1916 );
1917 pub fn atg__native_batch_norm_legit_no_stats_out(
1918 out__: *mut *mut C_tensor,
1919 out_: *mut C_tensor,
1920 save_mean_: *mut C_tensor,
1921 save_invstd_: *mut C_tensor,
1922 input_: *mut C_tensor,
1923 weight_: *mut C_tensor,
1924 bias_: *mut C_tensor,
1925 training_: c_int,
1926 momentum_: f64,
1927 eps_: f64,
1928 );
1929 pub fn atg__native_batch_norm_legit_no_training(
1930 out__: *mut *mut C_tensor,
1931 input_: *mut C_tensor,
1932 weight_: *mut C_tensor,
1933 bias_: *mut C_tensor,
1934 running_mean_: *mut C_tensor,
1935 running_var_: *mut C_tensor,
1936 momentum_: f64,
1937 eps_: f64,
1938 );
1939 pub fn atg__native_batch_norm_legit_no_training_out(
1940 out__: *mut *mut C_tensor,
1941 out0_: *mut C_tensor,
1942 out1_: *mut C_tensor,
1943 out2_: *mut C_tensor,
1944 input_: *mut C_tensor,
1945 weight_: *mut C_tensor,
1946 bias_: *mut C_tensor,
1947 running_mean_: *mut C_tensor,
1948 running_var_: *mut C_tensor,
1949 momentum_: f64,
1950 eps_: f64,
1951 );
1952 pub fn atg__native_batch_norm_legit_out(
1953 out__: *mut *mut C_tensor,
1954 out_: *mut C_tensor,
1955 save_mean_: *mut C_tensor,
1956 save_invstd_: *mut C_tensor,
1957 input_: *mut C_tensor,
1958 weight_: *mut C_tensor,
1959 bias_: *mut C_tensor,
1960 running_mean_: *mut C_tensor,
1961 running_var_: *mut C_tensor,
1962 training_: c_int,
1963 momentum_: f64,
1964 eps_: f64,
1965 );
1966 pub fn atg__native_multi_head_attention(
1967 out__: *mut *mut C_tensor,
1968 query_: *mut C_tensor,
1969 key_: *mut C_tensor,
1970 value_: *mut C_tensor,
1971 embed_dim_: i64,
1972 num_head_: i64,
1973 qkv_weight_: *mut C_tensor,
1974 qkv_bias_: *mut C_tensor,
1975 proj_weight_: *mut C_tensor,
1976 proj_bias_: *mut C_tensor,
1977 mask_: *mut C_tensor,
1978 need_weights_: c_int,
1979 average_attn_weights_: c_int,
1980 mask_type_v: i64,
1981 mask_type_null: i8,
1982 );
1983 pub fn atg__native_multi_head_attention_out(
1984 out__: *mut *mut C_tensor,
1985 out0_: *mut C_tensor,
1986 out1_: *mut C_tensor,
1987 query_: *mut C_tensor,
1988 key_: *mut C_tensor,
1989 value_: *mut C_tensor,
1990 embed_dim_: i64,
1991 num_head_: i64,
1992 qkv_weight_: *mut C_tensor,
1993 qkv_bias_: *mut C_tensor,
1994 proj_weight_: *mut C_tensor,
1995 proj_bias_: *mut C_tensor,
1996 mask_: *mut C_tensor,
1997 need_weights_: c_int,
1998 average_attn_weights_: c_int,
1999 mask_type_v: i64,
2000 mask_type_null: i8,
2001 );
2002 pub fn atg__neg_view(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2003 pub fn atg__neg_view_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2004 pub fn atg__neg_view_copy_out(
2005 out__: *mut *mut C_tensor,
2006 out_: *mut C_tensor,
2007 self_: *mut C_tensor,
2008 );
2009 pub fn atg__nested_compute_contiguous_strides_offsets(
2010 out__: *mut *mut C_tensor,
2011 nested_size_: *mut C_tensor,
2012 );
2013 pub fn atg__nested_from_padded(
2014 out__: *mut *mut C_tensor,
2015 padded_: *mut C_tensor,
2016 cpu_nested_shape_example_: *mut C_tensor,
2017 fuse_transform_0213_: c_int,
2018 );
2019 pub fn atg__nested_from_padded_and_nested_example(
2020 out__: *mut *mut C_tensor,
2021 padded_: *mut C_tensor,
2022 nt_example_: *mut C_tensor,
2023 );
2024 pub fn atg__nested_from_padded_and_nested_example_out(
2025 out__: *mut *mut C_tensor,
2026 out_: *mut C_tensor,
2027 padded_: *mut C_tensor,
2028 nt_example_: *mut C_tensor,
2029 );
2030 pub fn atg__nested_from_padded_out(
2031 out__: *mut *mut C_tensor,
2032 out_: *mut C_tensor,
2033 padded_: *mut C_tensor,
2034 cpu_nested_shape_example_: *mut C_tensor,
2035 fuse_transform_0213_: c_int,
2036 );
2037 pub fn atg__nested_from_padded_tensor(
2038 out__: *mut *mut C_tensor,
2039 padded_: *mut C_tensor,
2040 offsets_: *mut C_tensor,
2041 dummy_: *mut C_tensor,
2042 ragged_idx_: i64,
2043 min_seqlen_: *mut C_tensor,
2044 max_seqlen_: *mut C_tensor,
2045 sum_S_v: i64,
2046 sum_S_null: i8,
2047 );
2048 pub fn atg__nested_get_jagged_dummy(out__: *mut *mut C_tensor, any_: *mut C_tensor);
2049 pub fn atg__nested_get_lengths(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2050 pub fn atg__nested_get_max_seqlen(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2051 pub fn atg__nested_get_min_seqlen(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2052 pub fn atg__nested_get_offsets(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2053 pub fn atg__nested_get_ragged_idx(self_: *mut C_tensor) -> i64;
2054 pub fn atg__nested_get_values(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2055 pub fn atg__nested_get_values_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2056 pub fn atg__nested_get_values_copy_out(
2057 out__: *mut *mut C_tensor,
2058 out_: *mut C_tensor,
2059 self_: *mut C_tensor,
2060 );
2061 pub fn atg__nested_select_backward(
2062 out__: *mut *mut C_tensor,
2063 grad_output_: *mut C_tensor,
2064 self_: *mut C_tensor,
2065 dim_: i64,
2066 index_: i64,
2067 );
2068 pub fn atg__nested_sum_backward(
2069 out__: *mut *mut C_tensor,
2070 grad_: *mut C_tensor,
2071 self_: *mut C_tensor,
2072 dim_data: *const i64,
2073 dim_len: c_int,
2074 keepdim_: c_int,
2075 );
2076 pub fn atg__nested_view_from_buffer(
2077 out__: *mut *mut C_tensor,
2078 self_: *mut C_tensor,
2079 nested_size_: *mut C_tensor,
2080 nested_strides_: *mut C_tensor,
2081 offsets_: *mut C_tensor,
2082 );
2083 pub fn atg__nested_view_from_buffer_copy(
2084 out__: *mut *mut C_tensor,
2085 self_: *mut C_tensor,
2086 nested_size_: *mut C_tensor,
2087 nested_strides_: *mut C_tensor,
2088 offsets_: *mut C_tensor,
2089 );
2090 pub fn atg__nested_view_from_buffer_copy_out(
2091 out__: *mut *mut C_tensor,
2092 out_: *mut C_tensor,
2093 self_: *mut C_tensor,
2094 nested_size_: *mut C_tensor,
2095 nested_strides_: *mut C_tensor,
2096 offsets_: *mut C_tensor,
2097 );
2098 pub fn atg__nested_view_from_jagged(
2099 out__: *mut *mut C_tensor,
2100 self_: *mut C_tensor,
2101 offsets_: *mut C_tensor,
2102 dummy_: *mut C_tensor,
2103 lengths_: *mut C_tensor,
2104 ragged_idx_: i64,
2105 min_seqlen_: *mut C_tensor,
2106 max_seqlen_: *mut C_tensor,
2107 );
2108 pub fn atg__nested_view_from_jagged_copy(
2109 out__: *mut *mut C_tensor,
2110 self_: *mut C_tensor,
2111 offsets_: *mut C_tensor,
2112 dummy_: *mut C_tensor,
2113 lengths_: *mut C_tensor,
2114 ragged_idx_: i64,
2115 min_seqlen_: *mut C_tensor,
2116 max_seqlen_: *mut C_tensor,
2117 );
2118 pub fn atg__nested_view_from_jagged_copy_out(
2119 out__: *mut *mut C_tensor,
2120 out_: *mut C_tensor,
2121 self_: *mut C_tensor,
2122 offsets_: *mut C_tensor,
2123 dummy_: *mut C_tensor,
2124 lengths_: *mut C_tensor,
2125 ragged_idx_: i64,
2126 min_seqlen_: *mut C_tensor,
2127 max_seqlen_: *mut C_tensor,
2128 );
2129 pub fn atg__new_zeros_with_same_feature_meta(
2130 out__: *mut *mut C_tensor,
2131 self_: *mut C_tensor,
2132 other_: *mut C_tensor,
2133 self_num_batch_dims_: i64,
2134 );
2135 pub fn atg__new_zeros_with_same_feature_meta_out(
2136 out__: *mut *mut C_tensor,
2137 out_: *mut C_tensor,
2138 self_: *mut C_tensor,
2139 other_: *mut C_tensor,
2140 self_num_batch_dims_: i64,
2141 );
2142 pub fn atg__nnpack_available() -> c_int;
2143 pub fn atg__nnpack_spatial_convolution(
2144 out__: *mut *mut C_tensor,
2145 input_: *mut C_tensor,
2146 weight_: *mut C_tensor,
2147 bias_: *mut C_tensor,
2148 padding_data: *const i64,
2149 padding_len: c_int,
2150 stride_data: *const i64,
2151 stride_len: c_int,
2152 );
2153 pub fn atg__nnpack_spatial_convolution_out(
2154 out__: *mut *mut C_tensor,
2155 out_: *mut C_tensor,
2156 input_: *mut C_tensor,
2157 weight_: *mut C_tensor,
2158 bias_: *mut C_tensor,
2159 padding_data: *const i64,
2160 padding_len: c_int,
2161 stride_data: *const i64,
2162 stride_len: c_int,
2163 );
2164 pub fn atg__nnz(self_: *mut C_tensor) -> i64;
2165 pub fn atg__pack_padded_sequence(
2166 out__: *mut *mut C_tensor,
2167 input_: *mut C_tensor,
2168 lengths_: *mut C_tensor,
2169 batch_first_: c_int,
2170 );
2171 pub fn atg__pack_padded_sequence_backward(
2172 out__: *mut *mut C_tensor,
2173 grad_: *mut C_tensor,
2174 input_size_data: *const i64,
2175 input_size_len: c_int,
2176 batch_sizes_: *mut C_tensor,
2177 batch_first_: c_int,
2178 );
2179 pub fn atg__pack_padded_sequence_out(
2180 out__: *mut *mut C_tensor,
2181 out0_: *mut C_tensor,
2182 out1_: *mut C_tensor,
2183 input_: *mut C_tensor,
2184 lengths_: *mut C_tensor,
2185 batch_first_: c_int,
2186 );
2187 pub fn atg__pad_circular(
2188 out__: *mut *mut C_tensor,
2189 self_: *mut C_tensor,
2190 pad_data: *const i64,
2191 pad_len: c_int,
2192 );
2193 pub fn atg__pad_enum(
2194 out__: *mut *mut C_tensor,
2195 self_: *mut C_tensor,
2196 pad_data: *const i64,
2197 pad_len: c_int,
2198 mode_: i64,
2199 value_v: f64,
2200 value_null: i8,
2201 );
2202 pub fn atg__pad_packed_sequence(
2203 out__: *mut *mut C_tensor,
2204 data_: *mut C_tensor,
2205 batch_sizes_: *mut C_tensor,
2206 batch_first_: c_int,
2207 padding_value_: *mut C_scalar,
2208 total_length_: i64,
2209 );
2210 pub fn atg__pdist_backward(
2211 out__: *mut *mut C_tensor,
2212 grad_: *mut C_tensor,
2213 self_: *mut C_tensor,
2214 p_: f64,
2215 pdist_: *mut C_tensor,
2216 );
2217 pub fn atg__pdist_backward_out(
2218 out__: *mut *mut C_tensor,
2219 out_: *mut C_tensor,
2220 grad_: *mut C_tensor,
2221 self_: *mut C_tensor,
2222 p_: f64,
2223 pdist_: *mut C_tensor,
2224 );
2225 pub fn atg__philox_key_fold_in(out__: *mut *mut C_tensor, key_: *mut C_tensor, data_: i64);
2226 pub fn atg__philox_key_split(out__: *mut *mut C_tensor, key_: *mut C_tensor, num_splits_: i64);
2227 pub fn atg__philox_normal(
2228 out__: *mut *mut C_tensor,
2229 self_: *mut C_tensor,
2230 key_: *mut C_tensor,
2231 mean_: f64,
2232 std_: f64,
2233 );
2234 pub fn atg__philox_normal_(
2235 out__: *mut *mut C_tensor,
2236 self_: *mut C_tensor,
2237 key_: *mut C_tensor,
2238 mean_: f64,
2239 std_: f64,
2240 );
2241 pub fn atg__philox_normal_out(
2242 out__: *mut *mut C_tensor,
2243 out_: *mut C_tensor,
2244 self_: *mut C_tensor,
2245 key_: *mut C_tensor,
2246 mean_: f64,
2247 std_: f64,
2248 );
2249 pub fn atg__philox_uniform(
2250 out__: *mut *mut C_tensor,
2251 self_: *mut C_tensor,
2252 key_: *mut C_tensor,
2253 low_: f64,
2254 high_: f64,
2255 );
2256 pub fn atg__philox_uniform_(
2257 out__: *mut *mut C_tensor,
2258 self_: *mut C_tensor,
2259 key_: *mut C_tensor,
2260 low_: f64,
2261 high_: f64,
2262 );
2263 pub fn atg__philox_uniform_out(
2264 out__: *mut *mut C_tensor,
2265 out_: *mut C_tensor,
2266 self_: *mut C_tensor,
2267 key_: *mut C_tensor,
2268 low_: f64,
2269 high_: f64,
2270 );
2271 pub fn atg__pin_memory(out__: *mut *mut C_tensor, self_: *mut C_tensor, device_: c_int);
2272 pub fn atg__pin_memory_out(
2273 out__: *mut *mut C_tensor,
2274 out_: *mut C_tensor,
2275 self_: *mut C_tensor,
2276 device_: c_int,
2277 );
2278 pub fn atg__prelu_kernel(
2279 out__: *mut *mut C_tensor,
2280 self_: *mut C_tensor,
2281 weight_: *mut C_tensor,
2282 );
2283 pub fn atg__prelu_kernel_backward(
2284 out__: *mut *mut C_tensor,
2285 grad_output_: *mut C_tensor,
2286 self_: *mut C_tensor,
2287 weight_: *mut C_tensor,
2288 );
2289 pub fn atg__print(s_ptr: *const u8, s_len: c_int);
2290 pub fn atg__propagate_xla_data(input_: *mut C_tensor, output_: *mut C_tensor);
2291 pub fn atg__remove_batch_dim(
2292 out__: *mut *mut C_tensor,
2293 self_: *mut C_tensor,
2294 level_: i64,
2295 batch_size_: i64,
2296 out_dim_: i64,
2297 );
2298 pub fn atg__reshape_alias(
2299 out__: *mut *mut C_tensor,
2300 self_: *mut C_tensor,
2301 size_data: *const i64,
2302 size_len: c_int,
2303 stride_data: *const i64,
2304 stride_len: c_int,
2305 );
2306 pub fn atg__reshape_alias_copy(
2307 out__: *mut *mut C_tensor,
2308 self_: *mut C_tensor,
2309 size_data: *const i64,
2310 size_len: c_int,
2311 stride_data: *const i64,
2312 stride_len: c_int,
2313 );
2314 pub fn atg__reshape_alias_copy_out(
2315 out__: *mut *mut C_tensor,
2316 out_: *mut C_tensor,
2317 self_: *mut C_tensor,
2318 size_data: *const i64,
2319 size_len: c_int,
2320 stride_data: *const i64,
2321 stride_len: c_int,
2322 );
2323 pub fn atg__reshape_copy(
2324 out__: *mut *mut C_tensor,
2325 self_: *mut C_tensor,
2326 size_data: *const i64,
2327 size_len: c_int,
2328 );
2329 pub fn atg__reshape_from_tensor(
2330 out__: *mut *mut C_tensor,
2331 self_: *mut C_tensor,
2332 shape_: *mut C_tensor,
2333 );
2334 pub fn atg__resize_output(
2335 out__: *mut *mut C_tensor,
2336 self_: *mut C_tensor,
2337 size_data: *const i64,
2338 size_len: c_int,
2339 device_: c_int,
2340 );
2341 pub fn atg__resize_output_(
2342 out__: *mut *mut C_tensor,
2343 self_: *mut C_tensor,
2344 size_data: *const i64,
2345 size_len: c_int,
2346 device_: c_int,
2347 );
2348 pub fn atg__resize_output_out(
2349 out__: *mut *mut C_tensor,
2350 out_: *mut C_tensor,
2351 self_: *mut C_tensor,
2352 size_data: *const i64,
2353 size_len: c_int,
2354 device_: c_int,
2355 );
2356 pub fn atg__rowwise_prune(
2357 out__: *mut *mut C_tensor,
2358 weight_: *mut C_tensor,
2359 mask_: *mut C_tensor,
2360 compressed_indices_dtype_: c_int,
2361 );
2362 pub fn atg__safe_softmax(
2363 out__: *mut *mut C_tensor,
2364 self_: *mut C_tensor,
2365 dim_: i64,
2366 dtype_: c_int,
2367 );
2368 pub fn atg__sample_dirichlet(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2369 pub fn atg__sample_dirichlet_out(
2370 out__: *mut *mut C_tensor,
2371 out_: *mut C_tensor,
2372 self_: *mut C_tensor,
2373 );
2374 pub fn atg__saturate_weight_to_fp16(out__: *mut *mut C_tensor, weight_: *mut C_tensor);
2375 pub fn atg__scaled_dot_product_attention_math(
2376 out__: *mut *mut C_tensor,
2377 query_: *mut C_tensor,
2378 key_: *mut C_tensor,
2379 value_: *mut C_tensor,
2380 attn_mask_: *mut C_tensor,
2381 dropout_p_: f64,
2382 is_causal_: c_int,
2383 dropout_mask_: *mut C_tensor,
2384 scale_v: f64,
2385 scale_null: i8,
2386 enable_gqa_: c_int,
2387 );
2388 pub fn atg__scaled_dot_product_attention_math_for_mps(
2389 out__: *mut *mut C_tensor,
2390 query_: *mut C_tensor,
2391 key_: *mut C_tensor,
2392 value_: *mut C_tensor,
2393 attn_mask_: *mut C_tensor,
2394 dropout_p_: f64,
2395 is_causal_: c_int,
2396 dropout_mask_: *mut C_tensor,
2397 scale_v: f64,
2398 scale_null: i8,
2399 enable_gqa_: c_int,
2400 );
2401 pub fn atg__scaled_dot_product_cudnn_attention_backward(
2402 out__: *mut *mut C_tensor,
2403 grad_out_: *mut C_tensor,
2404 query_: *mut C_tensor,
2405 key_: *mut C_tensor,
2406 value_: *mut C_tensor,
2407 out_: *mut C_tensor,
2408 logsumexp_: *mut C_tensor,
2409 philox_seed_: *mut C_tensor,
2410 philox_offset_: *mut C_tensor,
2411 attn_bias_: *mut C_tensor,
2412 cum_seq_q_: *mut C_tensor,
2413 cum_seq_k_: *mut C_tensor,
2414 max_q_: i64,
2415 max_k_: i64,
2416 dropout_p_: f64,
2417 is_causal_: c_int,
2418 scale_v: f64,
2419 scale_null: i8,
2420 );
2421 pub fn atg__scaled_dot_product_efficient_attention(
2422 out__: *mut *mut C_tensor,
2423 query_: *mut C_tensor,
2424 key_: *mut C_tensor,
2425 value_: *mut C_tensor,
2426 attn_bias_: *mut C_tensor,
2427 compute_log_sumexp_: c_int,
2428 dropout_p_: f64,
2429 is_causal_: c_int,
2430 scale_v: f64,
2431 scale_null: i8,
2432 );
2433 pub fn atg__scaled_dot_product_flash_attention_backward(
2434 out__: *mut *mut C_tensor,
2435 grad_out_: *mut C_tensor,
2436 query_: *mut C_tensor,
2437 key_: *mut C_tensor,
2438 value_: *mut C_tensor,
2439 out_: *mut C_tensor,
2440 logsumexp_: *mut C_tensor,
2441 cum_seq_q_: *mut C_tensor,
2442 cum_seq_k_: *mut C_tensor,
2443 max_q_: i64,
2444 max_k_: i64,
2445 dropout_p_: f64,
2446 is_causal_: c_int,
2447 philox_seed_: *mut C_tensor,
2448 philox_offset_: *mut C_tensor,
2449 scale_v: f64,
2450 scale_null: i8,
2451 );
2452 pub fn atg__scaled_dot_product_flash_attention_for_cpu(
2453 out__: *mut *mut C_tensor,
2454 query_: *mut C_tensor,
2455 key_: *mut C_tensor,
2456 value_: *mut C_tensor,
2457 dropout_p_: f64,
2458 is_causal_: c_int,
2459 attn_mask_: *mut C_tensor,
2460 scale_v: f64,
2461 scale_null: i8,
2462 );
2463 pub fn atg__scaled_dot_product_flash_attention_for_cpu_backward(
2464 out__: *mut *mut C_tensor,
2465 grad_out_: *mut C_tensor,
2466 query_: *mut C_tensor,
2467 key_: *mut C_tensor,
2468 value_: *mut C_tensor,
2469 out_: *mut C_tensor,
2470 logsumexp_: *mut C_tensor,
2471 dropout_p_: f64,
2472 is_causal_: c_int,
2473 attn_mask_: *mut C_tensor,
2474 scale_v: f64,
2475 scale_null: i8,
2476 );
2477 pub fn atg__scaled_grouped_mm(
2478 out__: *mut *mut C_tensor,
2479 self_: *mut C_tensor,
2480 mat2_: *mut C_tensor,
2481 scale_a_: *mut C_tensor,
2482 scale_b_: *mut C_tensor,
2483 offs_: *mut C_tensor,
2484 bias_: *mut C_tensor,
2485 scale_result_: *mut C_tensor,
2486 out_dtype_: c_int,
2487 use_fast_accum_: c_int,
2488 );
2489 pub fn atg__scaled_grouped_mm_v2(
2490 out__: *mut *mut C_tensor,
2491 self_: *mut C_tensor,
2492 mat2_: *mut C_tensor,
2493 scale_a_data: *const *mut C_tensor,
2494 scale_a_len: c_int,
2495 recipe_a_data: *const i64,
2496 recipe_a_len: c_int,
2497 swizzle_a_data: *const i64,
2498 swizzle_a_len: c_int,
2499 scale_b_data: *const *mut C_tensor,
2500 scale_b_len: c_int,
2501 recipe_b_data: *const i64,
2502 recipe_b_len: c_int,
2503 swizzle_b_data: *const i64,
2504 swizzle_b_len: c_int,
2505 offs_: *mut C_tensor,
2506 bias_: *mut C_tensor,
2507 out_dtype_: c_int,
2508 contraction_dim_data: *const i64,
2509 contraction_dim_len: c_int,
2510 use_fast_accum_: c_int,
2511 );
2512 pub fn atg__scaled_mm(
2513 out__: *mut *mut C_tensor,
2514 self_: *mut C_tensor,
2515 mat2_: *mut C_tensor,
2516 scale_a_: *mut C_tensor,
2517 scale_b_: *mut C_tensor,
2518 bias_: *mut C_tensor,
2519 scale_result_: *mut C_tensor,
2520 out_dtype_: c_int,
2521 use_fast_accum_: c_int,
2522 );
2523 pub fn atg__scaled_mm_out(
2524 out__: *mut *mut C_tensor,
2525 out_: *mut C_tensor,
2526 self_: *mut C_tensor,
2527 mat2_: *mut C_tensor,
2528 scale_a_: *mut C_tensor,
2529 scale_b_: *mut C_tensor,
2530 bias_: *mut C_tensor,
2531 scale_result_: *mut C_tensor,
2532 out_dtype_: c_int,
2533 use_fast_accum_: c_int,
2534 );
2535 pub fn atg__scaled_mm_v2(
2536 out__: *mut *mut C_tensor,
2537 self_: *mut C_tensor,
2538 mat2_: *mut C_tensor,
2539 scale_a_data: *const *mut C_tensor,
2540 scale_a_len: c_int,
2541 recipe_a_data: *const i64,
2542 recipe_a_len: c_int,
2543 swizzle_a_data: *const i64,
2544 swizzle_a_len: c_int,
2545 scale_b_data: *const *mut C_tensor,
2546 scale_b_len: c_int,
2547 recipe_b_data: *const i64,
2548 recipe_b_len: c_int,
2549 swizzle_b_data: *const i64,
2550 swizzle_b_len: c_int,
2551 bias_: *mut C_tensor,
2552 out_dtype_: c_int,
2553 contraction_dim_data: *const i64,
2554 contraction_dim_len: c_int,
2555 use_fast_accum_: c_int,
2556 );
2557 pub fn atg__scaled_mm_v2_out(
2558 out__: *mut *mut C_tensor,
2559 out_: *mut C_tensor,
2560 self_: *mut C_tensor,
2561 mat2_: *mut C_tensor,
2562 scale_a_data: *const *mut C_tensor,
2563 scale_a_len: c_int,
2564 recipe_a_data: *const i64,
2565 recipe_a_len: c_int,
2566 swizzle_a_data: *const i64,
2567 swizzle_a_len: c_int,
2568 scale_b_data: *const *mut C_tensor,
2569 scale_b_len: c_int,
2570 recipe_b_data: *const i64,
2571 recipe_b_len: c_int,
2572 swizzle_b_data: *const i64,
2573 swizzle_b_len: c_int,
2574 bias_: *mut C_tensor,
2575 out_dtype_: c_int,
2576 contraction_dim_data: *const i64,
2577 contraction_dim_len: c_int,
2578 use_fast_accum_: c_int,
2579 );
2580 pub fn atg__scatter_reduce(
2581 out__: *mut *mut C_tensor,
2582 self_: *mut C_tensor,
2583 dim_: i64,
2584 index_: *mut C_tensor,
2585 src_: *mut C_tensor,
2586 reduce_ptr: *const u8,
2587 reduce_len: c_int,
2588 include_self_: c_int,
2589 );
2590 pub fn atg__scatter_reduce_(
2591 out__: *mut *mut C_tensor,
2592 self_: *mut C_tensor,
2593 dim_: i64,
2594 index_: *mut C_tensor,
2595 src_: *mut C_tensor,
2596 reduce_ptr: *const u8,
2597 reduce_len: c_int,
2598 include_self_: c_int,
2599 );
2600 pub fn atg__scatter_reduce_two_out(
2601 out__: *mut *mut C_tensor,
2602 out_: *mut C_tensor,
2603 self_: *mut C_tensor,
2604 dim_: i64,
2605 index_: *mut C_tensor,
2606 src_: *mut C_tensor,
2607 reduce_ptr: *const u8,
2608 reduce_len: c_int,
2609 include_self_: c_int,
2610 );
2611 pub fn atg__segment_reduce_backward(
2612 out__: *mut *mut C_tensor,
2613 grad_: *mut C_tensor,
2614 output_: *mut C_tensor,
2615 data_: *mut C_tensor,
2616 reduce_ptr: *const u8,
2617 reduce_len: c_int,
2618 lengths_: *mut C_tensor,
2619 offsets_: *mut C_tensor,
2620 axis_: i64,
2621 initial_: *mut C_scalar,
2622 );
2623 pub fn atg__segment_reduce_backward_out(
2624 out__: *mut *mut C_tensor,
2625 out_: *mut C_tensor,
2626 grad_: *mut C_tensor,
2627 output_: *mut C_tensor,
2628 data_: *mut C_tensor,
2629 reduce_ptr: *const u8,
2630 reduce_len: c_int,
2631 lengths_: *mut C_tensor,
2632 offsets_: *mut C_tensor,
2633 axis_: i64,
2634 initial_: *mut C_scalar,
2635 );
2636 pub fn atg__shape_as_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor);
2637 pub fn atg__slow_conv2d_backward(
2638 out__: *mut *mut C_tensor,
2639 grad_input_: *mut C_tensor,
2640 grad_weight_: *mut C_tensor,
2641 grad_bias_: *mut C_tensor,
2642 grad_output_: *mut C_tensor,
2643 self_: *mut C_tensor,
2644 weight_: *mut C_tensor,
2645 kernel_size_data: *const i64,
2646 kernel_size_len: c_int,
2647 stride_data: *const i64,
2648 stride_len: c_int,
2649 padding_data: *const i64,
2650 padding_len: c_int,
2651 );
2652 pub fn atg__sobol_engine_draw(
2653 out__: *mut *mut C_tensor,
2654 quasi_: *mut C_tensor,
2655 n_: i64,
2656 sobolstate_: *mut C_tensor,
2657 dimension_: i64,
2658 num_generated_: i64,
2659 dtype_: c_int,
2660 );
2661 pub fn atg__sobol_engine_ff_(
2662 out__: *mut *mut C_tensor,
2663 self_: *mut C_tensor,
2664 n_: i64,
2665 sobolstate_: *mut C_tensor,
2666 dimension_: i64,
2667 num_generated_: i64,
2668 );
2669 pub fn atg__sobol_engine_initialize_state_(
2670 out__: *mut *mut C_tensor,
2671 self_: *mut C_tensor,
2672 dimension_: i64,
2673 );
2674 pub fn atg__sobol_engine_scramble_(
2675 out__: *mut *mut C_tensor,
2676 self_: *mut C_tensor,
2677 ltm_: *mut C_tensor,
2678 dimension_: i64,
2679 );
2680 pub fn atg__softmax(
2681 out__: *mut *mut C_tensor,
2682 self_: *mut C_tensor,
2683 dim_: i64,
2684 half_to_float_: c_int,
2685 );
2686 pub fn atg__softmax_backward_data(
2687 out__: *mut *mut C_tensor,
2688 grad_output_: *mut C_tensor,
2689 output_: *mut C_tensor,
2690 dim_: i64,
2691 input_dtype_: c_int,
2692 );
2693 pub fn atg__softmax_backward_data_out(
2694 out__: *mut *mut C_tensor,
2695 grad_input_: *mut C_tensor,
2696 grad_output_: *mut C_tensor,
2697 output_: *mut C_tensor,
2698 dim_: i64,
2699 input_dtype_: c_int,
2700 );
2701 pub fn atg__softmax_out(
2702 out__: *mut *mut C_tensor,
2703 out_: *mut C_tensor,
2704 self_: *mut C_tensor,
2705 dim_: i64,
2706 half_to_float_: c_int,
2707 );
2708 pub fn atg__sparse_addmm(
2709 out__: *mut *mut C_tensor,
2710 self_: *mut C_tensor,
2711 mat1_: *mut C_tensor,
2712 mat2_: *mut C_tensor,
2713 );
2714 pub fn atg__sparse_addmm_out(
2715 out__: *mut *mut C_tensor,
2716 out_: *mut C_tensor,
2717 self_: *mut C_tensor,
2718 mat1_: *mut C_tensor,
2719 mat2_: *mut C_tensor,
2720 );
2721 pub fn atg__sparse_broadcast_to(
2722 out__: *mut *mut C_tensor,
2723 self_: *mut C_tensor,
2724 size_data: *const i64,
2725 size_len: c_int,
2726 );
2727 pub fn atg__sparse_broadcast_to_copy(
2728 out__: *mut *mut C_tensor,
2729 self_: *mut C_tensor,
2730 size_data: *const i64,
2731 size_len: c_int,
2732 );
2733 pub fn atg__sparse_broadcast_to_copy_out(
2734 out__: *mut *mut C_tensor,
2735 out_: *mut C_tensor,
2736 self_: *mut C_tensor,
2737 size_data: *const i64,
2738 size_len: c_int,
2739 );
2740 pub fn atg__sparse_bsc_tensor_unsafe(
2741 out__: *mut *mut C_tensor,
2742 ccol_indices_: *mut C_tensor,
2743 row_indices_: *mut C_tensor,
2744 values_: *mut C_tensor,
2745 size_data: *const i64,
2746 size_len: c_int,
2747 options_kind: c_int,
2748 options_device: c_int,
2749 );
2750 pub fn atg__sparse_bsr_tensor_unsafe(
2751 out__: *mut *mut C_tensor,
2752 crow_indices_: *mut C_tensor,
2753 col_indices_: *mut C_tensor,
2754 values_: *mut C_tensor,
2755 size_data: *const i64,
2756 size_len: c_int,
2757 options_kind: c_int,
2758 options_device: c_int,
2759 );
2760 pub fn atg__sparse_compressed_tensor_unsafe(
2761 out__: *mut *mut C_tensor,
2762 compressed_indices_: *mut C_tensor,
2763 plain_indices_: *mut C_tensor,
2764 values_: *mut C_tensor,
2765 size_data: *const i64,
2766 size_len: c_int,
2767 options_kind: c_int,
2768 options_device: c_int,
2769 );
2770 pub fn atg__sparse_compressed_tensor_with_dims(
2771 out__: *mut *mut C_tensor,
2772 nnz_: i64,
2773 dense_dim_: i64,
2774 size_data: *const i64,
2775 size_len: c_int,
2776 blocksize_data: *const i64,
2777 blocksize_len: c_int,
2778 index_dtype_: c_int,
2779 options_kind: c_int,
2780 options_device: c_int,
2781 );
2782 pub fn atg__sparse_coo_tensor_unsafe(
2783 out__: *mut *mut C_tensor,
2784 indices_: *mut C_tensor,
2785 values_: *mut C_tensor,
2786 size_data: *const i64,
2787 size_len: c_int,
2788 options_kind: c_int,
2789 options_device: c_int,
2790 is_coalesced_: c_int,
2791 );
2792 pub fn atg__sparse_coo_tensor_with_dims(
2793 out__: *mut *mut C_tensor,
2794 sparse_dim_: i64,
2795 dense_dim_: i64,
2796 size_data: *const i64,
2797 size_len: c_int,
2798 options_kind: c_int,
2799 options_device: c_int,
2800 );
2801 pub fn atg__sparse_coo_tensor_with_dims_and_tensors(
2802 out__: *mut *mut C_tensor,
2803 sparse_dim_: i64,
2804 dense_dim_: i64,
2805 size_data: *const i64,
2806 size_len: c_int,
2807 indices_: *mut C_tensor,
2808 values_: *mut C_tensor,
2809 options_kind: c_int,
2810 options_device: c_int,
2811 is_coalesced_: c_int,
2812 );
2813 pub fn atg__sparse_coo_tensor_with_dims_and_tensors_out(
2814 out__: *mut *mut C_tensor,
2815 out_: *mut C_tensor,
2816 sparse_dim_: i64,
2817 dense_dim_: i64,
2818 size_data: *const i64,
2819 size_len: c_int,
2820 indices_: *mut C_tensor,
2821 values_: *mut C_tensor,
2822 is_coalesced_: c_int,
2823 );
2824 pub fn atg__sparse_coo_tensor_with_dims_out(
2825 out__: *mut *mut C_tensor,
2826 out_: *mut C_tensor,
2827 sparse_dim_: i64,
2828 dense_dim_: i64,
2829 size_data: *const i64,
2830 size_len: c_int,
2831 );
2832 pub fn atg__sparse_csc_tensor_unsafe(
2833 out__: *mut *mut C_tensor,
2834 ccol_indices_: *mut C_tensor,
2835 row_indices_: *mut C_tensor,
2836 values_: *mut C_tensor,
2837 size_data: *const i64,
2838 size_len: c_int,
2839 options_kind: c_int,
2840 options_device: c_int,
2841 );
2842 pub fn atg__sparse_csr_prod(
2843 out__: *mut *mut C_tensor,
2844 self_: *mut C_tensor,
2845 dim_data: *const i64,
2846 dim_len: c_int,
2847 keepdim_: c_int,
2848 dtype_: c_int,
2849 );
2850 pub fn atg__sparse_csr_prod_dim_dtype_out(
2851 out__: *mut *mut C_tensor,
2852 out_: *mut C_tensor,
2853 self_: *mut C_tensor,
2854 dim_data: *const i64,
2855 dim_len: c_int,
2856 keepdim_: c_int,
2857 dtype_: c_int,
2858 );
2859 pub fn atg__sparse_csr_sum(
2860 out__: *mut *mut C_tensor,
2861 self_: *mut C_tensor,
2862 dim_data: *const i64,
2863 dim_len: c_int,
2864 keepdim_: c_int,
2865 dtype_: c_int,
2866 );
2867 pub fn atg__sparse_csr_sum_dim_dtype_out(
2868 out__: *mut *mut C_tensor,
2869 out_: *mut C_tensor,
2870 self_: *mut C_tensor,
2871 dim_data: *const i64,
2872 dim_len: c_int,
2873 keepdim_: c_int,
2874 dtype_: c_int,
2875 );
2876 pub fn atg__sparse_csr_tensor_unsafe(
2877 out__: *mut *mut C_tensor,
2878 crow_indices_: *mut C_tensor,
2879 col_indices_: *mut C_tensor,
2880 values_: *mut C_tensor,
2881 size_data: *const i64,
2882 size_len: c_int,
2883 options_kind: c_int,
2884 options_device: c_int,
2885 );
2886 pub fn atg__sparse_log_softmax(
2887 out__: *mut *mut C_tensor,
2888 self_: *mut C_tensor,
2889 dim_: i64,
2890 half_to_float_: c_int,
2891 );
2892 pub fn atg__sparse_log_softmax_backward_data(
2893 out__: *mut *mut C_tensor,
2894 grad_output_: *mut C_tensor,
2895 output_: *mut C_tensor,
2896 dim_: i64,
2897 self_: *mut C_tensor,
2898 );
2899 pub fn atg__sparse_log_softmax_backward_data_out(
2900 out__: *mut *mut C_tensor,
2901 out_: *mut C_tensor,
2902 grad_output_: *mut C_tensor,
2903 output_: *mut C_tensor,
2904 dim_: i64,
2905 self_: *mut C_tensor,
2906 );
2907 pub fn atg__sparse_log_softmax_int(
2908 out__: *mut *mut C_tensor,
2909 self_: *mut C_tensor,
2910 dim_: i64,
2911 dtype_: c_int,
2912 );
2913 pub fn atg__sparse_log_softmax_out(
2914 out__: *mut *mut C_tensor,
2915 out_: *mut C_tensor,
2916 self_: *mut C_tensor,
2917 dim_: i64,
2918 half_to_float_: c_int,
2919 );
2920 pub fn atg__sparse_mask_projection(
2921 out__: *mut *mut C_tensor,
2922 self_: *mut C_tensor,
2923 mask_: *mut C_tensor,
2924 accumulate_matches_: c_int,
2925 );
2926 pub fn atg__sparse_mask_projection_out(
2927 out__: *mut *mut C_tensor,
2928 out_: *mut C_tensor,
2929 self_: *mut C_tensor,
2930 mask_: *mut C_tensor,
2931 accumulate_matches_: c_int,
2932 );
2933 pub fn atg__sparse_mm(out__: *mut *mut C_tensor, sparse_: *mut C_tensor, dense_: *mut C_tensor);
2934 pub fn atg__sparse_mm_reduce(
2935 out__: *mut *mut C_tensor,
2936 sparse_: *mut C_tensor,
2937 dense_: *mut C_tensor,
2938 reduce_ptr: *const u8,
2939 reduce_len: c_int,
2940 );
2941 pub fn atg__sparse_mm_reduce_impl(
2942 out__: *mut *mut C_tensor,
2943 self_: *mut C_tensor,
2944 other_: *mut C_tensor,
2945 reduce_ptr: *const u8,
2946 reduce_len: c_int,
2947 );
2948 pub fn atg__sparse_semi_structured_apply(
2949 out__: *mut *mut C_tensor,
2950 input_: *mut C_tensor,
2951 thread_masks_: *mut C_tensor,
2952 );
2953 pub fn atg__sparse_semi_structured_apply_dense(
2954 out__: *mut *mut C_tensor,
2955 input_: *mut C_tensor,
2956 thread_masks_: *mut C_tensor,
2957 );
2958 pub fn atg__sparse_semi_structured_linear(
2959 out__: *mut *mut C_tensor,
2960 input_: *mut C_tensor,
2961 weight_: *mut C_tensor,
2962 meta_: *mut C_tensor,
2963 bias_: *mut C_tensor,
2964 activation_ptr: *const u8,
2965 activation_len: c_int,
2966 out_dtype_: c_int,
2967 );
2968 pub fn atg__sparse_semi_structured_mm(
2969 out__: *mut *mut C_tensor,
2970 mat1_: *mut C_tensor,
2971 mat1_meta_: *mut C_tensor,
2972 mat2_: *mut C_tensor,
2973 out_dtype_: c_int,
2974 );
2975 pub fn atg__sparse_semi_structured_tile(
2976 out__: *mut *mut C_tensor,
2977 input_: *mut C_tensor,
2978 algorithm_ptr: *const u8,
2979 algorithm_len: c_int,
2980 use_cutlass_: c_int,
2981 );
2982 pub fn atg__sparse_softmax(
2983 out__: *mut *mut C_tensor,
2984 self_: *mut C_tensor,
2985 dim_: i64,
2986 half_to_float_: c_int,
2987 );
2988 pub fn atg__sparse_softmax_backward_data(
2989 out__: *mut *mut C_tensor,
2990 grad_output_: *mut C_tensor,
2991 output_: *mut C_tensor,
2992 dim_: i64,
2993 self_: *mut C_tensor,
2994 );
2995 pub fn atg__sparse_softmax_backward_data_out(
2996 out__: *mut *mut C_tensor,
2997 out_: *mut C_tensor,
2998 grad_output_: *mut C_tensor,
2999 output_: *mut C_tensor,
3000 dim_: i64,
3001 self_: *mut C_tensor,
3002 );
3003 pub fn atg__sparse_softmax_int(
3004 out__: *mut *mut C_tensor,
3005 self_: *mut C_tensor,
3006 dim_: i64,
3007 dtype_: c_int,
3008 );
3009 pub fn atg__sparse_softmax_out(
3010 out__: *mut *mut C_tensor,
3011 out_: *mut C_tensor,
3012 self_: *mut C_tensor,
3013 dim_: i64,
3014 half_to_float_: c_int,
3015 );
3016 pub fn atg__sparse_sparse_matmul(
3017 out__: *mut *mut C_tensor,
3018 self_: *mut C_tensor,
3019 other_: *mut C_tensor,
3020 );
3021 pub fn atg__sparse_sparse_matmul_out(
3022 out__: *mut *mut C_tensor,
3023 out_: *mut C_tensor,
3024 self_: *mut C_tensor,
3025 other_: *mut C_tensor,
3026 );
3027 pub fn atg__sparse_sum(out__: *mut *mut C_tensor, self_: *mut C_tensor);
3028 pub fn atg__sparse_sum_backward(
3029 out__: *mut *mut C_tensor,
3030 grad_: *mut C_tensor,
3031 self_: *mut C_tensor,
3032 dim_data: *const i64,
3033 dim_len: c_int,
3034 );
3035 pub fn atg__sparse_sum_backward_out(
3036 out__: *mut *mut C_tensor,
3037 out_: *mut C_tensor,
3038 grad_: *mut C_tensor,
3039 self_: *mut C_tensor,
3040 dim_data: *const i64,
3041 dim_len: c_int,
3042 );
3043 pub fn atg__sparse_sum_dim(
3044 out__: *mut *mut C_tensor,
3045 self_: *mut C_tensor,
3046 dim_data: *const i64,
3047 dim_len: c_int,
3048 );
3049 pub fn atg__sparse_sum_dim_dtype(
3050 out__: *mut *mut C_tensor,
3051 self_: *mut C_tensor,
3052 dim_data: *const i64,
3053 dim_len: c_int,
3054 dtype_: c_int,
3055 );
3056 pub fn atg__sparse_sum_dim_out(
3057 out__: *mut *mut C_tensor,
3058 out_: *mut C_tensor,
3059 self_: *mut C_tensor,
3060 dim_data: *const i64,
3061 dim_len: c_int,
3062 );
3063 pub fn atg__sparse_sum_dtype(out__: *mut *mut C_tensor, self_: *mut C_tensor, dtype_: c_int);
3064 pub fn atg__spdiags(
3065 out__: *mut *mut C_tensor,
3066 diagonals_: *mut C_tensor,
3067 offsets_: *mut C_tensor,
3068 shape_data: *const i64,
3069 shape_len: c_int,
3070 layout_: i8,
3071 );
3072 pub fn atg__spdiags_out(
3073 out__: *mut *mut C_tensor,
3074 out_: *mut C_tensor,
3075 diagonals_: *mut C_tensor,
3076 offsets_: *mut C_tensor,
3077 shape_data: *const i64,
3078 shape_len: c_int,
3079 layout_: i8,
3080 );
3081 pub fn atg__spsolve(
3082 out__: *mut *mut C_tensor,
3083 A_: *mut C_tensor,
3084 B_: *mut C_tensor,
3085 left_: c_int,
3086 );
3087 pub fn atg__stack(
3088 out__: *mut *mut C_tensor,
3089 tensors_data: *const *mut C_tensor,
3090 tensors_len: c_int,
3091 dim_: i64,
3092 );
3093 pub fn atg__stack_out(
3094 out__: *mut *mut C_tensor,
3095 out_: *mut C_tensor,
3096 tensors_data: *const *mut C_tensor,
3097 tensors_len: c_int,
3098 dim_: i64,
3099 );
3100 pub fn atg__standard_gamma(out__: *mut *mut C_tensor, self_: *mut C_tensor);
3101 pub fn atg__standard_gamma_grad(
3102 out__: *mut *mut C_tensor,
3103 self_: *mut C_tensor,
3104 output_: *mut C_tensor,
3105 );
3106 pub fn atg__standard_gamma_grad_out(
3107 out__: *mut *mut C_tensor,
3108 out_: *mut C_tensor,
3109 self_: *mut C_tensor,
3110 output_: *mut C_tensor,
3111 );
3112 pub fn atg__standard_gamma_out(
3113 out__: *mut *mut C_tensor,
3114 out_: *mut C_tensor,
3115 self_: *mut C_tensor,
3116 );
3117 pub fn atg__test_ambiguous_defaults(
3118 out__: *mut *mut C_tensor,
3119 dummy_: *mut C_tensor,
3120 a_: i64,
3121 b_: i64,
3122 );
3123 pub fn atg__test_ambiguous_defaults_b(
3124 out__: *mut *mut C_tensor,
3125 dummy_: *mut C_tensor,
3126 a_: i64,
3127 b_ptr: *const u8,
3128 b_len: c_int,
3129 );
3130 pub fn atg__test_autograd_multiple_dispatch(out__: *mut *mut C_tensor, self_: *mut C_tensor);
3131 pub fn atg__test_autograd_multiple_dispatch_fullcoverage_out(
3132 out__: *mut *mut C_tensor,
3133 out_: *mut C_tensor,
3134 self_: *mut C_tensor,
3135 );
3136 pub fn atg__test_autograd_multiple_dispatch_ntonly(
3137 out__: *mut *mut C_tensor,
3138 self_: *mut C_tensor,
3139 b_: c_int,
3140 );
3141 pub fn atg__test_autograd_multiple_dispatch_view(
3142 out__: *mut *mut C_tensor,
3143 self_: *mut C_tensor,
3144 );
3145 pub fn atg__test_autograd_multiple_dispatch_view_copy(
3146 out__: *mut *mut C_tensor,
3147 self_: *mut C_tensor,
3148 );
3149 pub fn atg__test_autograd_multiple_dispatch_view_copy_out(
3150 out__: *mut *mut C_tensor,
3151 out_: *mut C_tensor,
3152 self_: *mut C_tensor,
3153 );
3154 pub fn atg__test_check_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor);
3155 pub fn atg__test_functorch_fallback(
3156 out__: *mut *mut C_tensor,
3157 self_: *mut C_tensor,
3158 other_: *mut C_tensor,
3159 );
3160 pub fn atg__test_functorch_fallback_out(
3161 out__: *mut *mut C_tensor,
3162 out_: *mut C_tensor,
3163 self_: *mut C_tensor,
3164 other_: *mut C_tensor,
3165 );
3166 pub fn atg__test_optional_filled_intlist(
3167 out__: *mut *mut C_tensor,
3168 values_: *mut C_tensor,
3169 addends_data: *const i64,
3170 addends_len: c_int,
3171 );
3172 pub fn atg__test_optional_filled_intlist_out(
3173 out__: *mut *mut C_tensor,
3174 out_: *mut C_tensor,
3175 values_: *mut C_tensor,
3176 addends_data: *const i64,
3177 addends_len: c_int,
3178 );
3179 pub fn atg__test_optional_floatlist(
3180 out__: *mut *mut C_tensor,
3181 values_: *mut C_tensor,
3182 addends_data: *const f64,
3183 addends_len: c_int,
3184 );
3185 pub fn atg__test_optional_floatlist_out(
3186 out__: *mut *mut C_tensor,
3187 out_: *mut C_tensor,
3188 values_: *mut C_tensor,
3189 addends_data: *const f64,
3190 addends_len: c_int,
3191 );
3192 pub fn atg__test_optional_intlist(
3193 out__: *mut *mut C_tensor,
3194 values_: *mut C_tensor,
3195 addends_data: *const i64,
3196 addends_len: c_int,
3197 );
3198 pub fn atg__test_optional_intlist_out(
3199 out__: *mut *mut C_tensor,
3200 out_: *mut C_tensor,
3201 values_: *mut C_tensor,
3202 addends_data: *const i64,
3203 addends_len: c_int,
3204 );
3205 pub fn atg__test_parallel_materialize(
3206 out__: *mut *mut C_tensor,
3207 self_: *mut C_tensor,
3208 num_parallel_: i64,
3209 skip_first_: c_int,
3210 );
3211 pub fn atg__test_serialization_subcmul(
3212 out__: *mut *mut C_tensor,
3213 self_: *mut C_tensor,
3214 other_: *mut C_tensor,
3215 );
3216 pub fn atg__test_string_default(
3217 out__: *mut *mut C_tensor,
3218 dummy_: *mut C_tensor,
3219 a_ptr: *const u8,
3220 a_len: c_int,
3221 b_ptr: *const u8,
3222 b_len: c_int,
3223 );
3224 pub fn atg__test_warn_in_autograd(out__: *mut *mut C_tensor, self_: *mut C_tensor);
3225 pub fn atg__test_warn_in_autograd_out(
3226 out__: *mut *mut C_tensor,
3227 out_: *mut C_tensor,
3228 self_: *mut C_tensor,
3229 );
3230 pub fn atg__to_copy(
3231 out__: *mut *mut C_tensor,
3232 self_: *mut C_tensor,
3233 options_kind: c_int,
3234 options_device: c_int,
3235 non_blocking_: c_int,
3236 );
3237 pub fn atg__to_copy_out(
3238 out__: *mut *mut C_tensor,
3239 out_: *mut C_tensor,
3240 self_: *mut C_tensor,
3241 non_blocking_: c_int,
3242 );
3243 pub fn atg__to_cpu(
3244 tensors_data: *const *mut C_tensor,
3245 tensors_len: c_int,
3246 ) -> *mut *mut C_tensor;
3247 pub fn atg__to_dense(
3248 out__: *mut *mut C_tensor,
3249 self_: *mut C_tensor,
3250 dtype_: c_int,
3251 masked_grad_: c_int,
3252 );
3253 pub fn atg__to_dense_out(
3254 out__: *mut *mut C_tensor,
3255 out_: *mut C_tensor,
3256 self_: *mut C_tensor,
3257 dtype_: c_int,
3258 masked_grad_: c_int,
3259 );
3260 pub fn atg__to_sparse(
3261 out__: *mut *mut C_tensor,
3262 self_: *mut C_tensor,
3263 layout_: i8,
3264 blocksize_data: *const i64,
3265 blocksize_len: c_int,
3266 dense_dim_v: i64,
3267 dense_dim_null: i8,
3268 );
3269 pub fn atg__to_sparse_bsc(
3270 out__: *mut *mut C_tensor,
3271 self_: *mut C_tensor,
3272 blocksize_data: *const i64,
3273 blocksize_len: c_int,
3274 dense_dim_v: i64,
3275 dense_dim_null: i8,
3276 );
3277 pub fn atg__to_sparse_bsc_out(
3278 out__: *mut *mut C_tensor,
3279 out_: *mut C_tensor,
3280 self_: *mut C_tensor,
3281 blocksize_data: *const i64,
3282 blocksize_len: c_int,
3283 dense_dim_v: i64,
3284 dense_dim_null: i8,
3285 );
3286 pub fn atg__to_sparse_bsr(
3287 out__: *mut *mut C_tensor,
3288 self_: *mut C_tensor,
3289 blocksize_data: *const i64,
3290 blocksize_len: c_int,
3291 dense_dim_v: i64,
3292 dense_dim_null: i8,
3293 );
3294 pub fn atg__to_sparse_bsr_out(
3295 out__: *mut *mut C_tensor,
3296 out_: *mut C_tensor,
3297 self_: *mut C_tensor,
3298 blocksize_data: *const i64,
3299 blocksize_len: c_int,
3300 dense_dim_v: i64,
3301 dense_dim_null: i8,
3302 );
3303 pub fn atg__to_sparse_csc(
3304 out__: *mut *mut C_tensor,
3305 self_: *mut C_tensor,
3306 dense_dim_v: i64,
3307 dense_dim_null: i8,
3308 );
3309 pub fn atg__to_sparse_csc_out(
3310 out__: *mut *mut C_tensor,
3311 out_: *mut C_tensor,
3312 self_: *mut C_tensor,
3313 dense_dim_v: i64,
3314 dense_dim_null: i8,
3315 );
3316 pub fn atg__to_sparse_csr(
3317 out__: *mut *mut C_tensor,
3318 self_: *mut C_tensor,
3319 dense_dim_v: i64,
3320 dense_dim_null: i8,
3321 );
3322 pub fn atg__to_sparse_csr_out(
3323 out__: *mut *mut C_tensor,
3324 out_: *mut C_tensor,
3325 self_: *mut C_tensor,
3326 dense_dim_v: i64,
3327 dense_dim_null: i8,
3328 );
3329 pub fn atg__to_sparse_out(
3330 out__: *mut *mut C_tensor,
3331 out_: *mut C_tensor,
3332 self_: *mut C_tensor,
3333 layout_: i8,
3334 blocksize_data: *const i64,
3335 blocksize_len: c_int,
3336 dense_dim_v: i64,
3337 dense_dim_null: i8,
3338 );
3339 pub fn atg__to_sparse_semi_structured(out__: *mut *mut C_tensor, dense_: *mut C_tensor);
3340 pub fn atg__to_sparse_sparse_dim(
3341 out__: *mut *mut C_tensor,
3342 self_: *mut C_tensor,
3343 sparse_dim_: i64,
3344 );
3345 pub fn atg__to_sparse_sparse_dim_out(
3346 out__: *mut *mut C_tensor,
3347 out_: *mut C_tensor,
3348 self_: *mut C_tensor,
3349 sparse_dim_: i64,
3350 );
3351 pub fn atg__transform_bias_rescale_qkv(
3352 out__: *mut *mut C_tensor,
3353 qkv_: *mut C_tensor,
3354 qkv_bias_: *mut C_tensor,
3355 num_heads_: i64,
3356 );
3357 pub fn atg__transform_bias_rescale_qkv_out(
3358 out__: *mut *mut C_tensor,
3359 out0_: *mut C_tensor,
3360 out1_: *mut C_tensor,
3361 out2_: *mut C_tensor,
3362 qkv_: *mut C_tensor,
3363 qkv_bias_: *mut C_tensor,
3364 num_heads_: i64,
3365 );
3366 pub fn atg__transformer_encoder_layer_fwd(
3367 out__: *mut *mut C_tensor,
3368 src_: *mut C_tensor,
3369 embed_dim_: i64,
3370 num_heads_: i64,
3371 qkv_weight_: *mut C_tensor,
3372 qkv_bias_: *mut C_tensor,
3373 proj_weight_: *mut C_tensor,
3374 proj_bias_: *mut C_tensor,
3375 use_gelu_: c_int,
3376 norm_first_: c_int,
3377 eps_: f64,
3378 norm_weight_1_: *mut C_tensor,
3379 norm_bias_1_: *mut C_tensor,
3380 norm_weight_2_: *mut C_tensor,
3381 norm_bias_2_: *mut C_tensor,
3382 ffn_weight_1_: *mut C_tensor,
3383 ffn_bias_1_: *mut C_tensor,
3384 ffn_weight_2_: *mut C_tensor,
3385 ffn_bias_2_: *mut C_tensor,
3386 mask_: *mut C_tensor,
3387 mask_type_v: i64,
3388 mask_type_null: i8,
3389 );
3390 pub fn atg__transformer_encoder_layer_fwd_out(
3391 out__: *mut *mut C_tensor,
3392 out_: *mut C_tensor,
3393 src_: *mut C_tensor,
3394 embed_dim_: i64,
3395 num_heads_: i64,
3396 qkv_weight_: *mut C_tensor,
3397 qkv_bias_: *mut C_tensor,
3398 proj_weight_: *mut C_tensor,
3399 proj_bias_: *mut C_tensor,
3400 use_gelu_: c_int,
3401 norm_first_: c_int,
3402 eps_: f64,
3403 norm_weight_1_: *mut C_tensor,
3404 norm_bias_1_: *mut C_tensor,
3405 norm_weight_2_: *mut C_tensor,
3406 norm_bias_2_: *mut C_tensor,
3407 ffn_weight_1_: *mut C_tensor,
3408 ffn_bias_1_: *mut C_tensor,
3409 ffn_weight_2_: *mut C_tensor,
3410 ffn_bias_2_: *mut C_tensor,
3411 mask_: *mut C_tensor,
3412 mask_type_v: i64,
3413 mask_type_null: i8,
3414 );
3415 pub fn atg__trilinear(
3416 out__: *mut *mut C_tensor,
3417 i1_: *mut C_tensor,
3418 i2_: *mut C_tensor,
3419 i3_: *mut C_tensor,
3420 expand1_data: *const i64,
3421 expand1_len: c_int,
3422 expand2_data: *const i64,
3423 expand2_len: c_int,
3424 expand3_data: *const i64,
3425 expand3_len: c_int,
3426 sumdim_data: *const i64,
3427 sumdim_len: c_int,
3428 unroll_dim_: i64,
3429 );
3430 pub fn atg__trilinear_out(
3431 out__: *mut *mut C_tensor,
3432 out_: *mut C_tensor,
3433 i1_: *mut C_tensor,
3434 i2_: *mut C_tensor,
3435 i3_: *mut C_tensor,
3436 expand1_data: *const i64,
3437 expand1_len: c_int,
3438 expand2_data: *const i64,
3439 expand2_len: c_int,
3440 expand3_data: *const i64,
3441 expand3_len: c_int,
3442 sumdim_data: *const i64,
3443 sumdim_len: c_int,
3444 unroll_dim_: i64,
3445 );
3446 pub fn atg__triton_multi_head_attention(
3447 out__: *mut *mut C_tensor,
3448 query_: *mut C_tensor,
3449 key_: *mut C_tensor,
3450 value_: *mut C_tensor,
3451 embed_dim_: i64,
3452 num_head_: i64,
3453 qkv_weight_: *mut C_tensor,
3454 qkv_bias_: *mut C_tensor,
3455 proj_weight_: *mut C_tensor,
3456 proj_bias_: *mut C_tensor,
3457 mask_: *mut C_tensor,
3458 );
3459 pub fn atg__triton_multi_head_attention_out(
3460 out__: *mut *mut C_tensor,
3461 out_: *mut C_tensor,
3462 query_: *mut C_tensor,
3463 key_: *mut C_tensor,
3464 value_: *mut C_tensor,
3465 embed_dim_: i64,
3466 num_head_: i64,
3467 qkv_weight_: *mut C_tensor,
3468 qkv_bias_: *mut C_tensor,
3469 proj_weight_: *mut C_tensor,
3470 proj_bias_: *mut C_tensor,
3471 mask_: *mut C_tensor,
3472 );
3473 pub fn atg__triton_scaled_dot_attention(
3474 out__: *mut *mut C_tensor,
3475 q_: *mut C_tensor,
3476 k_: *mut C_tensor,
3477 v_: *mut C_tensor,
3478 dropout_p_: f64,
3479 );
3480 pub fn atg__triton_scaled_dot_attention_out(
3481 out__: *mut *mut C_tensor,
3482 out_: *mut C_tensor,
3483 q_: *mut C_tensor,
3484 k_: *mut C_tensor,
3485 v_: *mut C_tensor,
3486 dropout_p_: f64,
3487 );
3488 pub fn atg__unique(
3489 out__: *mut *mut C_tensor,
3490 self_: *mut C_tensor,
3491 sorted_: c_int,
3492 return_inverse_: c_int,
3493 );
3494 pub fn atg__unique2(
3495 out__: *mut *mut C_tensor,
3496 self_: *mut C_tensor,
3497 sorted_: c_int,
3498 return_inverse_: c_int,
3499 return_counts_: c_int,
3500 );
3501 pub fn atg__unique2_out(
3502 out__: *mut *mut C_tensor,
3503 out0_: *mut C_tensor,
3504 out1_: *mut C_tensor,
3505 out2_: *mut C_tensor,
3506 self_: *mut C_tensor,
3507 sorted_: c_int,
3508 return_inverse_: c_int,
3509 return_counts_: c_int,
3510 );
3511 pub fn atg__unique_out(
3512 out__: *mut *mut C_tensor,
3513 out0_: *mut C_tensor,
3514 out1_: *mut C_tensor,
3515 self_: *mut C_tensor,
3516 sorted_: c_int,
3517 return_inverse_: c_int,
3518 );
3519 pub fn atg__unpack_dual(out__: *mut *mut C_tensor, dual_: *mut C_tensor, level_: i64);
3520 pub fn atg__unsafe_index(
3521 out__: *mut *mut C_tensor,
3522 self_: *mut C_tensor,
3523 indices_data: *const *mut C_tensor,
3524 indices_len: c_int,
3525 );
3526 pub fn atg__unsafe_index_put(
3527 out__: *mut *mut C_tensor,
3528 self_: *mut C_tensor,
3529 indices_data: *const *mut C_tensor,
3530 indices_len: c_int,
3531 values_: *mut C_tensor,
3532 accumulate_: c_int,
3533 );
3534 pub fn atg__unsafe_masked_index(
3535 out__: *mut *mut C_tensor,
3536 self_: *mut C_tensor,
3537 mask_: *mut C_tensor,
3538 indices_data: *const *mut C_tensor,
3539 indices_len: c_int,
3540 fill_: *mut C_scalar,
3541 );
3542 pub fn atg__unsafe_masked_index_put_accumulate(
3543 out__: *mut *mut C_tensor,
3544 self_: *mut C_tensor,
3545 mask_: *mut C_tensor,
3546 indices_data: *const *mut C_tensor,
3547 indices_len: c_int,
3548 values_: *mut C_tensor,
3549 );
3550 pub fn atg__unsafe_view(
3551 out__: *mut *mut C_tensor,
3552 self_: *mut C_tensor,
3553 size_data: *const i64,
3554 size_len: c_int,
3555 );
3556 pub fn atg__unsafe_view_out(
3557 out__: *mut *mut C_tensor,
3558 out_: *mut C_tensor,
3559 self_: *mut C_tensor,
3560 size_data: *const i64,
3561 size_len: c_int,
3562 );
3563 pub fn atg__upsample_bicubic2d_aa(
3564 out__: *mut *mut C_tensor,
3565 self_: *mut C_tensor,
3566 output_size_data: *const i64,
3567 output_size_len: c_int,
3568 align_corners_: c_int,
3569 scales_h_v: f64,
3570 scales_h_null: i8,
3571 scales_w_v: f64,
3572 scales_w_null: i8,
3573 );
3574 pub fn atg__upsample_bicubic2d_aa_backward(
3575 out__: *mut *mut C_tensor,
3576 grad_output_: *mut C_tensor,
3577 output_size_data: *const i64,
3578 output_size_len: c_int,
3579 input_size_data: *const i64,
3580 input_size_len: c_int,
3581 align_corners_: c_int,
3582 scales_h_v: f64,
3583 scales_h_null: i8,
3584 scales_w_v: f64,
3585 scales_w_null: i8,
3586 );
3587 pub fn atg__upsample_bicubic2d_aa_backward_grad_input(
3588 out__: *mut *mut C_tensor,
3589 grad_input_: *mut C_tensor,
3590 grad_output_: *mut C_tensor,
3591 output_size_data: *const i64,
3592 output_size_len: c_int,
3593 input_size_data: *const i64,
3594 input_size_len: c_int,
3595 align_corners_: c_int,
3596 scales_h_v: f64,
3597 scales_h_null: i8,
3598 scales_w_v: f64,
3599 scales_w_null: i8,
3600 );
3601 pub fn atg__upsample_bicubic2d_aa_out(
3602 out__: *mut *mut C_tensor,
3603 out_: *mut C_tensor,
3604 self_: *mut C_tensor,
3605 output_size_data: *const i64,
3606 output_size_len: c_int,
3607 align_corners_: c_int,
3608 scales_h_v: f64,
3609 scales_h_null: i8,
3610 scales_w_v: f64,
3611 scales_w_null: i8,
3612 );
3613 pub fn atg__upsample_bicubic2d_aa_vec(
3614 out__: *mut *mut C_tensor,
3615 input_: *mut C_tensor,
3616 output_size_data: *const i64,
3617 output_size_len: c_int,
3618 align_corners_: c_int,
3619 scale_factors_data: *const f64,
3620 scale_factors_len: c_int,
3621 );
3622 pub fn atg__upsample_bilinear2d_aa(
3623 out__: *mut *mut C_tensor,
3624 self_: *mut C_tensor,
3625 output_size_data: *const i64,
3626 output_size_len: c_int,
3627 align_corners_: c_int,
3628 scales_h_v: f64,
3629 scales_h_null: i8,
3630 scales_w_v: f64,
3631 scales_w_null: i8,
3632 );
3633 pub fn atg__upsample_bilinear2d_aa_backward(
3634 out__: *mut *mut C_tensor,
3635 grad_output_: *mut C_tensor,
3636 output_size_data: *const i64,
3637 output_size_len: c_int,
3638 input_size_data: *const i64,
3639 input_size_len: c_int,
3640 align_corners_: c_int,
3641 scales_h_v: f64,
3642 scales_h_null: i8,
3643 scales_w_v: f64,
3644 scales_w_null: i8,
3645 );
3646 pub fn atg__upsample_bilinear2d_aa_backward_grad_input(
3647 out__: *mut *mut C_tensor,
3648 grad_input_: *mut C_tensor,
3649 grad_output_: *mut C_tensor,
3650 output_size_data: *const i64,
3651 output_size_len: c_int,
3652 input_size_data: *const i64,
3653 input_size_len: c_int,
3654 align_corners_: c_int,
3655 scales_h_v: f64,
3656 scales_h_null: i8,
3657 scales_w_v: f64,
3658 scales_w_null: i8,
3659 );
3660 pub fn atg__upsample_bilinear2d_aa_out(
3661 out__: *mut *mut C_tensor,
3662 out_: *mut C_tensor,
3663 self_: *mut C_tensor,
3664 output_size_data: *const i64,
3665 output_size_len: c_int,
3666 align_corners_: c_int,
3667 scales_h_v: f64,
3668 scales_h_null: i8,
3669 scales_w_v: f64,
3670 scales_w_null: i8,
3671 );
3672 pub fn atg__upsample_bilinear2d_aa_vec(
3673 out__: *mut *mut C_tensor,
3674 input_: *mut C_tensor,
3675 output_size_data: *const i64,
3676 output_size_len: c_int,
3677 align_corners_: c_int,
3678 scale_factors_data: *const f64,
3679 scale_factors_len: c_int,
3680 );
3681 pub fn atg__upsample_lanczos2d_aa(
3682 out__: *mut *mut C_tensor,
3683 self_: *mut C_tensor,
3684 output_size_data: *const i64,
3685 output_size_len: c_int,
3686 align_corners_: c_int,
3687 scales_h_v: f64,
3688 scales_h_null: i8,
3689 scales_w_v: f64,
3690 scales_w_null: i8,
3691 );
3692 pub fn atg__upsample_lanczos2d_aa_backward(
3693 out__: *mut *mut C_tensor,
3694 grad_output_: *mut C_tensor,
3695 output_size_data: *const i64,
3696 output_size_len: c_int,
3697 input_size_data: *const i64,
3698 input_size_len: c_int,
3699 align_corners_: c_int,
3700 scales_h_v: f64,
3701 scales_h_null: i8,
3702 scales_w_v: f64,
3703 scales_w_null: i8,
3704 );
3705 pub fn atg__upsample_lanczos2d_aa_backward_grad_input(
3706 out__: *mut *mut C_tensor,
3707 grad_input_: *mut C_tensor,
3708 grad_output_: *mut C_tensor,
3709 output_size_data: *const i64,
3710 output_size_len: c_int,
3711 input_size_data: *const i64,
3712 input_size_len: c_int,
3713 align_corners_: c_int,
3714 scales_h_v: f64,
3715 scales_h_null: i8,
3716 scales_w_v: f64,
3717 scales_w_null: i8,
3718 );
3719 pub fn atg__upsample_lanczos2d_aa_out(
3720 out__: *mut *mut C_tensor,
3721 out_: *mut C_tensor,
3722 self_: *mut C_tensor,
3723 output_size_data: *const i64,
3724 output_size_len: c_int,
3725 align_corners_: c_int,
3726 scales_h_v: f64,
3727 scales_h_null: i8,
3728 scales_w_v: f64,
3729 scales_w_null: i8,
3730 );
3731 pub fn atg__upsample_lanczos2d_aa_vec(
3732 out__: *mut *mut C_tensor,
3733 input_: *mut C_tensor,
3734 output_size_data: *const i64,
3735 output_size_len: c_int,
3736 align_corners_: c_int,
3737 scale_factors_data: *const f64,
3738 scale_factors_len: c_int,
3739 );
3740 pub fn atg__upsample_nearest_exact1d(
3741 out__: *mut *mut C_tensor,
3742 self_: *mut C_tensor,
3743 output_size_data: *const i64,
3744 output_size_len: c_int,
3745 scales_v: f64,
3746 scales_null: i8,
3747 );
3748 pub fn atg__upsample_nearest_exact1d_backward(
3749 out__: *mut *mut C_tensor,
3750 grad_output_: *mut C_tensor,
3751 output_size_data: *const i64,
3752 output_size_len: c_int,
3753 input_size_data: *const i64,
3754 input_size_len: c_int,
3755 scales_v: f64,
3756 scales_null: i8,
3757 );
3758 pub fn atg__upsample_nearest_exact1d_backward_grad_input(
3759 out__: *mut *mut C_tensor,
3760 grad_input_: *mut C_tensor,
3761 grad_output_: *mut C_tensor,
3762 output_size_data: *const i64,
3763 output_size_len: c_int,
3764 input_size_data: *const i64,
3765 input_size_len: c_int,
3766 scales_v: f64,
3767 scales_null: i8,
3768 );
3769 pub fn atg__upsample_nearest_exact1d_out(
3770 out__: *mut *mut C_tensor,
3771 out_: *mut C_tensor,
3772 self_: *mut C_tensor,
3773 output_size_data: *const i64,
3774 output_size_len: c_int,
3775 scales_v: f64,
3776 scales_null: i8,
3777 );
3778 pub fn atg__upsample_nearest_exact1d_vec(
3779 out__: *mut *mut C_tensor,
3780 input_: *mut C_tensor,
3781 output_size_data: *const i64,
3782 output_size_len: c_int,
3783 scale_factors_data: *const f64,
3784 scale_factors_len: c_int,
3785 );
3786 pub fn atg__upsample_nearest_exact2d(
3787 out__: *mut *mut C_tensor,
3788 self_: *mut C_tensor,
3789 output_size_data: *const i64,
3790 output_size_len: c_int,
3791 scales_h_v: f64,
3792 scales_h_null: i8,
3793 scales_w_v: f64,
3794 scales_w_null: i8,
3795 );
3796 pub fn atg__upsample_nearest_exact2d_backward(
3797 out__: *mut *mut C_tensor,
3798 grad_output_: *mut C_tensor,
3799 output_size_data: *const i64,
3800 output_size_len: c_int,
3801 input_size_data: *const i64,
3802 input_size_len: c_int,
3803 scales_h_v: f64,
3804 scales_h_null: i8,
3805 scales_w_v: f64,
3806 scales_w_null: i8,
3807 );
3808 pub fn atg__upsample_nearest_exact2d_backward_grad_input(
3809 out__: *mut *mut C_tensor,
3810 grad_input_: *mut C_tensor,
3811 grad_output_: *mut C_tensor,
3812 output_size_data: *const i64,
3813 output_size_len: c_int,
3814 input_size_data: *const i64,
3815 input_size_len: c_int,
3816 scales_h_v: f64,
3817 scales_h_null: i8,
3818 scales_w_v: f64,
3819 scales_w_null: i8,
3820 );
3821 pub fn atg__upsample_nearest_exact2d_out(
3822 out__: *mut *mut C_tensor,
3823 out_: *mut C_tensor,
3824 self_: *mut C_tensor,
3825 output_size_data: *const i64,
3826 output_size_len: c_int,
3827 scales_h_v: f64,
3828 scales_h_null: i8,
3829 scales_w_v: f64,
3830 scales_w_null: i8,
3831 );
3832 pub fn atg__upsample_nearest_exact2d_vec(
3833 out__: *mut *mut C_tensor,
3834 input_: *mut C_tensor,
3835 output_size_data: *const i64,
3836 output_size_len: c_int,
3837 scale_factors_data: *const f64,
3838 scale_factors_len: c_int,
3839 );
3840 pub fn atg__upsample_nearest_exact3d(
3841 out__: *mut *mut C_tensor,
3842 self_: *mut C_tensor,
3843 output_size_data: *const i64,
3844 output_size_len: c_int,
3845 scales_d_v: f64,
3846 scales_d_null: i8,
3847 scales_h_v: f64,
3848 scales_h_null: i8,
3849 scales_w_v: f64,
3850 scales_w_null: i8,
3851 );
3852 pub fn atg__upsample_nearest_exact3d_backward(
3853 out__: *mut *mut C_tensor,
3854 grad_output_: *mut C_tensor,
3855 output_size_data: *const i64,
3856 output_size_len: c_int,
3857 input_size_data: *const i64,
3858 input_size_len: c_int,
3859 scales_d_v: f64,
3860 scales_d_null: i8,
3861 scales_h_v: f64,
3862 scales_h_null: i8,
3863 scales_w_v: f64,
3864 scales_w_null: i8,
3865 );
3866 pub fn atg__upsample_nearest_exact3d_backward_grad_input(
3867 out__: *mut *mut C_tensor,
3868 grad_input_: *mut C_tensor,
3869 grad_output_: *mut C_tensor,
3870 output_size_data: *const i64,
3871 output_size_len: c_int,
3872 input_size_data: *const i64,
3873 input_size_len: c_int,
3874 scales_d_v: f64,
3875 scales_d_null: i8,
3876 scales_h_v: f64,
3877 scales_h_null: i8,
3878 scales_w_v: f64,
3879 scales_w_null: i8,
3880 );
3881 pub fn atg__upsample_nearest_exact3d_out(
3882 out__: *mut *mut C_tensor,
3883 out_: *mut C_tensor,
3884 self_: *mut C_tensor,
3885 output_size_data: *const i64,
3886 output_size_len: c_int,
3887 scales_d_v: f64,
3888 scales_d_null: i8,
3889 scales_h_v: f64,
3890 scales_h_null: i8,
3891 scales_w_v: f64,
3892 scales_w_null: i8,
3893 );
3894 pub fn atg__upsample_nearest_exact3d_vec(
3895 out__: *mut *mut C_tensor,
3896 input_: *mut C_tensor,
3897 output_size_data: *const i64,
3898 output_size_len: c_int,
3899 scale_factors_data: *const f64,
3900 scale_factors_len: c_int,
3901 );
3902 pub fn atg__use_cudnn_ctc_loss(
3903 log_probs_: *mut C_tensor,
3904 targets_: *mut C_tensor,
3905 input_lengths_data: *const i64,
3906 input_lengths_len: c_int,
3907 target_lengths_data: *const i64,
3908 target_lengths_len: c_int,
3909 blank_: i64,
3910 ) -> c_int;
3911 pub fn atg__use_cudnn_ctc_loss_tensor(
3912 log_probs_: *mut C_tensor,
3913 targets_: *mut C_tensor,
3914 input_lengths_: *mut C_tensor,
3915 target_lengths_: *mut C_tensor,
3916 blank_: i64,
3917 ) -> c_int;
3918 pub fn atg__use_cudnn_rnn_flatten_weight() -> c_int;
3919 pub fn atg__use_miopen_ctc_loss(
3920 log_probs_: *mut C_tensor,
3921 targets_: *mut C_tensor,
3922 input_lengths_data: *const i64,
3923 input_lengths_len: c_int,
3924 target_lengths_data: *const i64,
3925 target_lengths_len: c_int,
3926 blank_: i64,
3927 ) -> c_int;
3928 pub fn atg__use_miopen_ctc_loss_tensor(
3929 log_probs_: *mut C_tensor,
3930 targets_: *mut C_tensor,
3931 input_lengths_: *mut C_tensor,
3932 target_lengths_: *mut C_tensor,
3933 blank_: i64,
3934 ) -> c_int;
3935 pub fn atg__validate_compressed_sparse_indices(
3936 is_crow_: c_int,
3937 compressed_idx_: *mut C_tensor,
3938 plain_idx_: *mut C_tensor,
3939 cdim_: i64,
3940 dim_: i64,
3941 nnz_: i64,
3942 );
3943 pub fn atg__validate_sparse_bsc_tensor_args(
3944 ccol_indices_: *mut C_tensor,
3945 row_indices_: *mut C_tensor,
3946 values_: *mut C_tensor,
3947 size_data: *const i64,
3948 size_len: c_int,
3949 check_pinning_: c_int,
3950 );
3951 pub fn atg__validate_sparse_bsr_tensor_args(
3952 crow_indices_: *mut C_tensor,
3953 col_indices_: *mut C_tensor,
3954 values_: *mut C_tensor,
3955 size_data: *const i64,
3956 size_len: c_int,
3957 check_pinning_: c_int,
3958 );
3959 pub fn atg__validate_sparse_compressed_tensor_args(
3960 compressed_indices_: *mut C_tensor,
3961 plain_indices_: *mut C_tensor,
3962 values_: *mut C_tensor,
3963 size_data: *const i64,
3964 size_len: c_int,
3965 layout_: i8,
3966 check_pinning_: c_int,
3967 );
3968 pub fn atg__validate_sparse_csc_tensor_args(
3969 ccol_indices_: *mut C_tensor,
3970 row_indices_: *mut C_tensor,
3971 values_: *mut C_tensor,
3972 size_data: *const i64,
3973 size_len: c_int,
3974 check_pinning_: c_int,
3975 );
3976 pub fn atg__validate_sparse_csr_tensor_args(
3977 crow_indices_: *mut C_tensor,
3978 col_indices_: *mut C_tensor,
3979 values_: *mut C_tensor,
3980 size_data: *const i64,
3981 size_len: c_int,
3982 check_pinning_: c_int,
3983 );
3984 pub fn atg__values(out__: *mut *mut C_tensor, self_: *mut C_tensor);
3985 pub fn atg__values_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
3986 pub fn atg__values_copy_out(
3987 out__: *mut *mut C_tensor,
3988 out_: *mut C_tensor,
3989 self_: *mut C_tensor,
3990 );
3991 pub fn atg__version(self_: *mut C_tensor) -> i64;
3992 pub fn atg__weight_int4pack_mm(
3993 out__: *mut *mut C_tensor,
3994 self_: *mut C_tensor,
3995 mat2_: *mut C_tensor,
3996 qGroupSize_: i64,
3997 qScaleAndZeros_: *mut C_tensor,
3998 );
3999 pub fn atg__weight_int4pack_mm_for_cpu(
4000 out__: *mut *mut C_tensor,
4001 self_: *mut C_tensor,
4002 mat2_: *mut C_tensor,
4003 qGroupSize_: i64,
4004 qScaleAndZeros_: *mut C_tensor,
4005 );
4006 pub fn atg__weight_int4pack_mm_with_scales_and_zeros(
4007 out__: *mut *mut C_tensor,
4008 self_: *mut C_tensor,
4009 mat2_: *mut C_tensor,
4010 qGroupSize_: i64,
4011 qScale_: *mut C_tensor,
4012 qZeros_: *mut C_tensor,
4013 );
4014 pub fn atg__weight_int8pack_mm(
4015 out__: *mut *mut C_tensor,
4016 self_: *mut C_tensor,
4017 mat2_: *mut C_tensor,
4018 scales_: *mut C_tensor,
4019 );
4020 pub fn atg__weight_norm(
4021 out__: *mut *mut C_tensor,
4022 v_: *mut C_tensor,
4023 g_: *mut C_tensor,
4024 dim_: i64,
4025 );
4026 pub fn atg__weight_norm_differentiable_backward(
4027 out__: *mut *mut C_tensor,
4028 grad_w_: *mut C_tensor,
4029 saved_v_: *mut C_tensor,
4030 saved_g_: *mut C_tensor,
4031 saved_norms_: *mut C_tensor,
4032 dim_: i64,
4033 );
4034 pub fn atg__weight_norm_interface(
4035 out__: *mut *mut C_tensor,
4036 v_: *mut C_tensor,
4037 g_: *mut C_tensor,
4038 dim_: i64,
4039 );
4040 pub fn atg__weight_norm_interface_backward(
4041 out__: *mut *mut C_tensor,
4042 grad_w_: *mut C_tensor,
4043 saved_v_: *mut C_tensor,
4044 saved_g_: *mut C_tensor,
4045 saved_norms_: *mut C_tensor,
4046 dim_: i64,
4047 );
4048 pub fn atg__weight_norm_interface_backward_out(
4049 out__: *mut *mut C_tensor,
4050 out0_: *mut C_tensor,
4051 out1_: *mut C_tensor,
4052 grad_w_: *mut C_tensor,
4053 saved_v_: *mut C_tensor,
4054 saved_g_: *mut C_tensor,
4055 saved_norms_: *mut C_tensor,
4056 dim_: i64,
4057 );
4058 pub fn atg__weight_norm_interface_out(
4059 out__: *mut *mut C_tensor,
4060 out0_: *mut C_tensor,
4061 out1_: *mut C_tensor,
4062 v_: *mut C_tensor,
4063 g_: *mut C_tensor,
4064 dim_: i64,
4065 );
4066 pub fn atg__wrapped_linear_prepack(
4067 out__: *mut *mut C_tensor,
4068 weight_: *mut C_tensor,
4069 weight_scale_: *mut C_tensor,
4070 weight_zero_point_: *mut C_tensor,
4071 bias_: *mut C_tensor,
4072 );
4073 pub fn atg__wrapped_quantized_linear_prepacked(
4074 out__: *mut *mut C_tensor,
4075 input_: *mut C_tensor,
4076 input_scale_: *mut C_tensor,
4077 input_zero_point_: *mut C_tensor,
4078 packed_weight_: *mut C_tensor,
4079 output_scale_: *mut C_tensor,
4080 output_zero_point_: *mut C_tensor,
4081 out_channel_: i64,
4082 );
4083 pub fn atg_abs(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4084 pub fn atg_abs_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4085 pub fn atg_abs_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4086 pub fn atg_absolute(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4087 pub fn atg_absolute_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4088 pub fn atg_absolute_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4089 pub fn atg_acos(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4090 pub fn atg_acos_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4091 pub fn atg_acos_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4092 pub fn atg_acosh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4093 pub fn atg_acosh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4094 pub fn atg_acosh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4095 pub fn atg_adaptive_avg_pool1d(
4096 out__: *mut *mut C_tensor,
4097 self_: *mut C_tensor,
4098 output_size_data: *const i64,
4099 output_size_len: c_int,
4100 );
4101 pub fn atg_adaptive_avg_pool1d_out(
4102 out__: *mut *mut C_tensor,
4103 out_: *mut C_tensor,
4104 self_: *mut C_tensor,
4105 output_size_data: *const i64,
4106 output_size_len: c_int,
4107 );
4108 pub fn atg_adaptive_avg_pool2d(
4109 out__: *mut *mut C_tensor,
4110 self_: *mut C_tensor,
4111 output_size_data: *const i64,
4112 output_size_len: c_int,
4113 );
4114 pub fn atg_adaptive_avg_pool2d_out(
4115 out__: *mut *mut C_tensor,
4116 out_: *mut C_tensor,
4117 self_: *mut C_tensor,
4118 output_size_data: *const i64,
4119 output_size_len: c_int,
4120 );
4121 pub fn atg_adaptive_avg_pool3d(
4122 out__: *mut *mut C_tensor,
4123 self_: *mut C_tensor,
4124 output_size_data: *const i64,
4125 output_size_len: c_int,
4126 );
4127 pub fn atg_adaptive_avg_pool3d_backward(
4128 out__: *mut *mut C_tensor,
4129 grad_input_: *mut C_tensor,
4130 grad_output_: *mut C_tensor,
4131 self_: *mut C_tensor,
4132 );
4133 pub fn atg_adaptive_avg_pool3d_out(
4134 out__: *mut *mut C_tensor,
4135 out_: *mut C_tensor,
4136 self_: *mut C_tensor,
4137 output_size_data: *const i64,
4138 output_size_len: c_int,
4139 );
4140 pub fn atg_adaptive_max_pool1d(
4141 out__: *mut *mut C_tensor,
4142 self_: *mut C_tensor,
4143 output_size_data: *const i64,
4144 output_size_len: c_int,
4145 );
4146 pub fn atg_adaptive_max_pool2d(
4147 out__: *mut *mut C_tensor,
4148 self_: *mut C_tensor,
4149 output_size_data: *const i64,
4150 output_size_len: c_int,
4151 );
4152 pub fn atg_adaptive_max_pool2d_backward(
4153 out__: *mut *mut C_tensor,
4154 grad_output_: *mut C_tensor,
4155 self_: *mut C_tensor,
4156 indices_: *mut C_tensor,
4157 );
4158 pub fn atg_adaptive_max_pool2d_backward_grad_input(
4159 out__: *mut *mut C_tensor,
4160 grad_input_: *mut C_tensor,
4161 grad_output_: *mut C_tensor,
4162 self_: *mut C_tensor,
4163 indices_: *mut C_tensor,
4164 );
4165 pub fn atg_adaptive_max_pool2d_out(
4166 out__: *mut *mut C_tensor,
4167 out_: *mut C_tensor,
4168 indices_: *mut C_tensor,
4169 self_: *mut C_tensor,
4170 output_size_data: *const i64,
4171 output_size_len: c_int,
4172 );
4173 pub fn atg_adaptive_max_pool3d(
4174 out__: *mut *mut C_tensor,
4175 self_: *mut C_tensor,
4176 output_size_data: *const i64,
4177 output_size_len: c_int,
4178 );
4179 pub fn atg_adaptive_max_pool3d_backward(
4180 out__: *mut *mut C_tensor,
4181 grad_output_: *mut C_tensor,
4182 self_: *mut C_tensor,
4183 indices_: *mut C_tensor,
4184 );
4185 pub fn atg_adaptive_max_pool3d_backward_grad_input(
4186 out__: *mut *mut C_tensor,
4187 grad_input_: *mut C_tensor,
4188 grad_output_: *mut C_tensor,
4189 self_: *mut C_tensor,
4190 indices_: *mut C_tensor,
4191 );
4192 pub fn atg_adaptive_max_pool3d_out(
4193 out__: *mut *mut C_tensor,
4194 out_: *mut C_tensor,
4195 indices_: *mut C_tensor,
4196 self_: *mut C_tensor,
4197 output_size_data: *const i64,
4198 output_size_len: c_int,
4199 );
4200 pub fn atg_add(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
4201 pub fn atg_add_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
4202 pub fn atg_add_out(
4203 out__: *mut *mut C_tensor,
4204 out_: *mut C_tensor,
4205 self_: *mut C_tensor,
4206 other_: *mut C_tensor,
4207 );
4208 pub fn atg_add_scalar(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
4209 pub fn atg_add_scalar_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
4210 pub fn atg_add_scalar_out(
4211 out__: *mut *mut C_tensor,
4212 out_: *mut C_tensor,
4213 self_: *mut C_tensor,
4214 other_: *mut C_scalar,
4215 );
4216 pub fn atg_addbmm(
4217 out__: *mut *mut C_tensor,
4218 self_: *mut C_tensor,
4219 batch1_: *mut C_tensor,
4220 batch2_: *mut C_tensor,
4221 );
4222 pub fn atg_addbmm_(
4223 out__: *mut *mut C_tensor,
4224 self_: *mut C_tensor,
4225 batch1_: *mut C_tensor,
4226 batch2_: *mut C_tensor,
4227 );
4228 pub fn atg_addbmm_out(
4229 out__: *mut *mut C_tensor,
4230 out_: *mut C_tensor,
4231 self_: *mut C_tensor,
4232 batch1_: *mut C_tensor,
4233 batch2_: *mut C_tensor,
4234 );
4235 pub fn atg_addcdiv(
4236 out__: *mut *mut C_tensor,
4237 self_: *mut C_tensor,
4238 tensor1_: *mut C_tensor,
4239 tensor2_: *mut C_tensor,
4240 );
4241 pub fn atg_addcdiv_(
4242 out__: *mut *mut C_tensor,
4243 self_: *mut C_tensor,
4244 tensor1_: *mut C_tensor,
4245 tensor2_: *mut C_tensor,
4246 );
4247 pub fn atg_addcdiv_out(
4248 out__: *mut *mut C_tensor,
4249 out_: *mut C_tensor,
4250 self_: *mut C_tensor,
4251 tensor1_: *mut C_tensor,
4252 tensor2_: *mut C_tensor,
4253 );
4254 pub fn atg_addcmul(
4255 out__: *mut *mut C_tensor,
4256 self_: *mut C_tensor,
4257 tensor1_: *mut C_tensor,
4258 tensor2_: *mut C_tensor,
4259 );
4260 pub fn atg_addcmul_(
4261 out__: *mut *mut C_tensor,
4262 self_: *mut C_tensor,
4263 tensor1_: *mut C_tensor,
4264 tensor2_: *mut C_tensor,
4265 );
4266 pub fn atg_addcmul_out(
4267 out__: *mut *mut C_tensor,
4268 out_: *mut C_tensor,
4269 self_: *mut C_tensor,
4270 tensor1_: *mut C_tensor,
4271 tensor2_: *mut C_tensor,
4272 );
4273 pub fn atg_addmm(
4274 out__: *mut *mut C_tensor,
4275 self_: *mut C_tensor,
4276 mat1_: *mut C_tensor,
4277 mat2_: *mut C_tensor,
4278 );
4279 pub fn atg_addmm_(
4280 out__: *mut *mut C_tensor,
4281 self_: *mut C_tensor,
4282 mat1_: *mut C_tensor,
4283 mat2_: *mut C_tensor,
4284 );
4285 pub fn atg_addmm_dtype(
4286 out__: *mut *mut C_tensor,
4287 self_: *mut C_tensor,
4288 mat1_: *mut C_tensor,
4289 mat2_: *mut C_tensor,
4290 out_dtype_: c_int,
4291 );
4292 pub fn atg_addmm_dtype_out(
4293 out__: *mut *mut C_tensor,
4294 out_: *mut C_tensor,
4295 self_: *mut C_tensor,
4296 mat1_: *mut C_tensor,
4297 mat2_: *mut C_tensor,
4298 out_dtype_: c_int,
4299 );
4300 pub fn atg_addmm_out(
4301 out__: *mut *mut C_tensor,
4302 out_: *mut C_tensor,
4303 self_: *mut C_tensor,
4304 mat1_: *mut C_tensor,
4305 mat2_: *mut C_tensor,
4306 );
4307 pub fn atg_addmv(
4308 out__: *mut *mut C_tensor,
4309 self_: *mut C_tensor,
4310 mat_: *mut C_tensor,
4311 vec_: *mut C_tensor,
4312 );
4313 pub fn atg_addmv_(
4314 out__: *mut *mut C_tensor,
4315 self_: *mut C_tensor,
4316 mat_: *mut C_tensor,
4317 vec_: *mut C_tensor,
4318 );
4319 pub fn atg_addmv_out(
4320 out__: *mut *mut C_tensor,
4321 out_: *mut C_tensor,
4322 self_: *mut C_tensor,
4323 mat_: *mut C_tensor,
4324 vec_: *mut C_tensor,
4325 );
4326 pub fn atg_addr(
4327 out__: *mut *mut C_tensor,
4328 self_: *mut C_tensor,
4329 vec1_: *mut C_tensor,
4330 vec2_: *mut C_tensor,
4331 );
4332 pub fn atg_addr_(
4333 out__: *mut *mut C_tensor,
4334 self_: *mut C_tensor,
4335 vec1_: *mut C_tensor,
4336 vec2_: *mut C_tensor,
4337 );
4338 pub fn atg_addr_out(
4339 out__: *mut *mut C_tensor,
4340 out_: *mut C_tensor,
4341 self_: *mut C_tensor,
4342 vec1_: *mut C_tensor,
4343 vec2_: *mut C_tensor,
4344 );
4345 pub fn atg_adjoint(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4346 pub fn atg_affine_grid_generator(
4347 out__: *mut *mut C_tensor,
4348 theta_: *mut C_tensor,
4349 size_data: *const i64,
4350 size_len: c_int,
4351 align_corners_: c_int,
4352 );
4353 pub fn atg_affine_grid_generator_backward(
4354 out__: *mut *mut C_tensor,
4355 grad_: *mut C_tensor,
4356 size_data: *const i64,
4357 size_len: c_int,
4358 align_corners_: c_int,
4359 );
4360 pub fn atg_affine_grid_generator_out(
4361 out__: *mut *mut C_tensor,
4362 out_: *mut C_tensor,
4363 theta_: *mut C_tensor,
4364 size_data: *const i64,
4365 size_len: c_int,
4366 align_corners_: c_int,
4367 );
4368 pub fn atg_alias(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4369 pub fn atg_alias_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4370 pub fn atg_alias_copy_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4371 pub fn atg_all(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4372 pub fn atg_all_all_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4373 pub fn atg_all_dim(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, keepdim_: c_int);
4374 pub fn atg_all_dims(
4375 out__: *mut *mut C_tensor,
4376 self_: *mut C_tensor,
4377 dim_data: *const i64,
4378 dim_len: c_int,
4379 keepdim_: c_int,
4380 );
4381 pub fn atg_all_dims_out(
4382 out__: *mut *mut C_tensor,
4383 out_: *mut C_tensor,
4384 self_: *mut C_tensor,
4385 dim_data: *const i64,
4386 dim_len: c_int,
4387 keepdim_: c_int,
4388 );
4389 pub fn atg_all_out(
4390 out__: *mut *mut C_tensor,
4391 out_: *mut C_tensor,
4392 self_: *mut C_tensor,
4393 dim_: i64,
4394 keepdim_: c_int,
4395 );
4396 pub fn atg_allclose(
4397 self_: *mut C_tensor,
4398 other_: *mut C_tensor,
4399 rtol_: f64,
4400 atol_: f64,
4401 equal_nan_: c_int,
4402 ) -> c_int;
4403 pub fn atg_alpha_dropout(
4404 out__: *mut *mut C_tensor,
4405 input_: *mut C_tensor,
4406 p_: f64,
4407 train_: c_int,
4408 );
4409 pub fn atg_alpha_dropout_(
4410 out__: *mut *mut C_tensor,
4411 self_: *mut C_tensor,
4412 p_: f64,
4413 train_: c_int,
4414 );
4415 pub fn atg_amax(
4416 out__: *mut *mut C_tensor,
4417 self_: *mut C_tensor,
4418 dim_data: *const i64,
4419 dim_len: c_int,
4420 keepdim_: c_int,
4421 );
4422 pub fn atg_amax_out(
4423 out__: *mut *mut C_tensor,
4424 out_: *mut C_tensor,
4425 self_: *mut C_tensor,
4426 dim_data: *const i64,
4427 dim_len: c_int,
4428 keepdim_: c_int,
4429 );
4430 pub fn atg_amin(
4431 out__: *mut *mut C_tensor,
4432 self_: *mut C_tensor,
4433 dim_data: *const i64,
4434 dim_len: c_int,
4435 keepdim_: c_int,
4436 );
4437 pub fn atg_amin_out(
4438 out__: *mut *mut C_tensor,
4439 out_: *mut C_tensor,
4440 self_: *mut C_tensor,
4441 dim_data: *const i64,
4442 dim_len: c_int,
4443 keepdim_: c_int,
4444 );
4445 pub fn atg_aminmax(
4446 out__: *mut *mut C_tensor,
4447 self_: *mut C_tensor,
4448 dim_v: i64,
4449 dim_null: i8,
4450 keepdim_: c_int,
4451 );
4452 pub fn atg_aminmax_out(
4453 out__: *mut *mut C_tensor,
4454 min_: *mut C_tensor,
4455 max_: *mut C_tensor,
4456 self_: *mut C_tensor,
4457 dim_v: i64,
4458 dim_null: i8,
4459 keepdim_: c_int,
4460 );
4461 pub fn atg_angle(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4462 pub fn atg_angle_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4463 pub fn atg_any(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4464 pub fn atg_any_all_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4465 pub fn atg_any_dim(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, keepdim_: c_int);
4466 pub fn atg_any_dims(
4467 out__: *mut *mut C_tensor,
4468 self_: *mut C_tensor,
4469 dim_data: *const i64,
4470 dim_len: c_int,
4471 keepdim_: c_int,
4472 );
4473 pub fn atg_any_dims_out(
4474 out__: *mut *mut C_tensor,
4475 out_: *mut C_tensor,
4476 self_: *mut C_tensor,
4477 dim_data: *const i64,
4478 dim_len: c_int,
4479 keepdim_: c_int,
4480 );
4481 pub fn atg_any_out(
4482 out__: *mut *mut C_tensor,
4483 out_: *mut C_tensor,
4484 self_: *mut C_tensor,
4485 dim_: i64,
4486 keepdim_: c_int,
4487 );
4488 pub fn atg_arange(
4489 out__: *mut *mut C_tensor,
4490 end_: *mut C_scalar,
4491 options_kind: c_int,
4492 options_device: c_int,
4493 );
4494 pub fn atg_arange_start(
4495 out__: *mut *mut C_tensor,
4496 start_: *mut C_scalar,
4497 end_: *mut C_scalar,
4498 options_kind: c_int,
4499 options_device: c_int,
4500 );
4501 pub fn atg_arange_start_step(
4502 out__: *mut *mut C_tensor,
4503 start_: *mut C_scalar,
4504 end_: *mut C_scalar,
4505 step_: *mut C_scalar,
4506 options_kind: c_int,
4507 options_device: c_int,
4508 );
4509 pub fn atg_arccos(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4510 pub fn atg_arccos_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4511 pub fn atg_arccos_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4512 pub fn atg_arccosh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4513 pub fn atg_arccosh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4514 pub fn atg_arccosh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4515 pub fn atg_arcsin(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4516 pub fn atg_arcsin_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4517 pub fn atg_arcsin_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4518 pub fn atg_arcsinh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4519 pub fn atg_arcsinh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4520 pub fn atg_arcsinh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4521 pub fn atg_arctan(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4522 pub fn atg_arctan2(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
4523 pub fn atg_arctan2_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
4524 pub fn atg_arctan2_out(
4525 out__: *mut *mut C_tensor,
4526 out_: *mut C_tensor,
4527 self_: *mut C_tensor,
4528 other_: *mut C_tensor,
4529 );
4530 pub fn atg_arctan_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4531 pub fn atg_arctan_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4532 pub fn atg_arctanh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4533 pub fn atg_arctanh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4534 pub fn atg_arctanh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4535 pub fn atg_argmax(
4536 out__: *mut *mut C_tensor,
4537 self_: *mut C_tensor,
4538 dim_v: i64,
4539 dim_null: i8,
4540 keepdim_: c_int,
4541 );
4542 pub fn atg_argmax_out(
4543 out__: *mut *mut C_tensor,
4544 out_: *mut C_tensor,
4545 self_: *mut C_tensor,
4546 dim_v: i64,
4547 dim_null: i8,
4548 keepdim_: c_int,
4549 );
4550 pub fn atg_argmin(
4551 out__: *mut *mut C_tensor,
4552 self_: *mut C_tensor,
4553 dim_v: i64,
4554 dim_null: i8,
4555 keepdim_: c_int,
4556 );
4557 pub fn atg_argmin_out(
4558 out__: *mut *mut C_tensor,
4559 out_: *mut C_tensor,
4560 self_: *mut C_tensor,
4561 dim_v: i64,
4562 dim_null: i8,
4563 keepdim_: c_int,
4564 );
4565 pub fn atg_argsort(
4566 out__: *mut *mut C_tensor,
4567 self_: *mut C_tensor,
4568 dim_: i64,
4569 descending_: c_int,
4570 );
4571 pub fn atg_argsort_stable(
4572 out__: *mut *mut C_tensor,
4573 self_: *mut C_tensor,
4574 stable_: c_int,
4575 dim_: i64,
4576 descending_: c_int,
4577 );
4578 pub fn atg_argsort_stable_out(
4579 out__: *mut *mut C_tensor,
4580 out_: *mut C_tensor,
4581 self_: *mut C_tensor,
4582 stable_: c_int,
4583 dim_: i64,
4584 descending_: c_int,
4585 );
4586 pub fn atg_argwhere(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4587 pub fn atg_as_strided(
4588 out__: *mut *mut C_tensor,
4589 self_: *mut C_tensor,
4590 size_data: *const i64,
4591 size_len: c_int,
4592 stride_data: *const i64,
4593 stride_len: c_int,
4594 storage_offset_v: i64,
4595 storage_offset_null: i8,
4596 );
4597 pub fn atg_as_strided_(
4598 out__: *mut *mut C_tensor,
4599 self_: *mut C_tensor,
4600 size_data: *const i64,
4601 size_len: c_int,
4602 stride_data: *const i64,
4603 stride_len: c_int,
4604 storage_offset_v: i64,
4605 storage_offset_null: i8,
4606 );
4607 pub fn atg_as_strided_copy(
4608 out__: *mut *mut C_tensor,
4609 self_: *mut C_tensor,
4610 size_data: *const i64,
4611 size_len: c_int,
4612 stride_data: *const i64,
4613 stride_len: c_int,
4614 storage_offset_v: i64,
4615 storage_offset_null: i8,
4616 );
4617 pub fn atg_as_strided_copy_out(
4618 out__: *mut *mut C_tensor,
4619 out_: *mut C_tensor,
4620 self_: *mut C_tensor,
4621 size_data: *const i64,
4622 size_len: c_int,
4623 stride_data: *const i64,
4624 stride_len: c_int,
4625 storage_offset_v: i64,
4626 storage_offset_null: i8,
4627 );
4628 pub fn atg_as_strided_scatter(
4629 out__: *mut *mut C_tensor,
4630 self_: *mut C_tensor,
4631 src_: *mut C_tensor,
4632 size_data: *const i64,
4633 size_len: c_int,
4634 stride_data: *const i64,
4635 stride_len: c_int,
4636 storage_offset_v: i64,
4637 storage_offset_null: i8,
4638 );
4639 pub fn atg_as_strided_scatter_out(
4640 out__: *mut *mut C_tensor,
4641 out_: *mut C_tensor,
4642 self_: *mut C_tensor,
4643 src_: *mut C_tensor,
4644 size_data: *const i64,
4645 size_len: c_int,
4646 stride_data: *const i64,
4647 stride_len: c_int,
4648 storage_offset_v: i64,
4649 storage_offset_null: i8,
4650 );
4651 pub fn atg_asin(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4652 pub fn atg_asin_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4653 pub fn atg_asin_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4654 pub fn atg_asinh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4655 pub fn atg_asinh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4656 pub fn atg_asinh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4657 pub fn atg_atan(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4658 pub fn atg_atan2(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
4659 pub fn atg_atan2_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
4660 pub fn atg_atan2_out(
4661 out__: *mut *mut C_tensor,
4662 out_: *mut C_tensor,
4663 self_: *mut C_tensor,
4664 other_: *mut C_tensor,
4665 );
4666 pub fn atg_atan_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4667 pub fn atg_atan_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4668 pub fn atg_atanh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4669 pub fn atg_atanh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4670 pub fn atg_atanh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
4671 pub fn atg_atleast_1d(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4672 pub fn atg_atleast_1d_sequence(
4673 tensors_data: *const *mut C_tensor,
4674 tensors_len: c_int,
4675 ) -> *mut *mut C_tensor;
4676 pub fn atg_atleast_2d(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4677 pub fn atg_atleast_2d_sequence(
4678 tensors_data: *const *mut C_tensor,
4679 tensors_len: c_int,
4680 ) -> *mut *mut C_tensor;
4681 pub fn atg_atleast_3d(out__: *mut *mut C_tensor, self_: *mut C_tensor);
4682 pub fn atg_atleast_3d_sequence(
4683 tensors_data: *const *mut C_tensor,
4684 tensors_len: c_int,
4685 ) -> *mut *mut C_tensor;
4686 pub fn atg_avg_pool1d(
4687 out__: *mut *mut C_tensor,
4688 self_: *mut C_tensor,
4689 kernel_size_data: *const i64,
4690 kernel_size_len: c_int,
4691 stride_data: *const i64,
4692 stride_len: c_int,
4693 padding_data: *const i64,
4694 padding_len: c_int,
4695 ceil_mode_: c_int,
4696 count_include_pad_: c_int,
4697 );
4698 pub fn atg_avg_pool1d_out(
4699 out__: *mut *mut C_tensor,
4700 out_: *mut C_tensor,
4701 self_: *mut C_tensor,
4702 kernel_size_data: *const i64,
4703 kernel_size_len: c_int,
4704 stride_data: *const i64,
4705 stride_len: c_int,
4706 padding_data: *const i64,
4707 padding_len: c_int,
4708 ceil_mode_: c_int,
4709 count_include_pad_: c_int,
4710 );
4711 pub fn atg_avg_pool2d(
4712 out__: *mut *mut C_tensor,
4713 self_: *mut C_tensor,
4714 kernel_size_data: *const i64,
4715 kernel_size_len: c_int,
4716 stride_data: *const i64,
4717 stride_len: c_int,
4718 padding_data: *const i64,
4719 padding_len: c_int,
4720 ceil_mode_: c_int,
4721 count_include_pad_: c_int,
4722 divisor_override_v: i64,
4723 divisor_override_null: i8,
4724 );
4725 pub fn atg_avg_pool2d_backward(
4726 out__: *mut *mut C_tensor,
4727 grad_output_: *mut C_tensor,
4728 self_: *mut C_tensor,
4729 kernel_size_data: *const i64,
4730 kernel_size_len: c_int,
4731 stride_data: *const i64,
4732 stride_len: c_int,
4733 padding_data: *const i64,
4734 padding_len: c_int,
4735 ceil_mode_: c_int,
4736 count_include_pad_: c_int,
4737 divisor_override_v: i64,
4738 divisor_override_null: i8,
4739 );
4740 pub fn atg_avg_pool2d_backward_grad_input(
4741 out__: *mut *mut C_tensor,
4742 grad_input_: *mut C_tensor,
4743 grad_output_: *mut C_tensor,
4744 self_: *mut C_tensor,
4745 kernel_size_data: *const i64,
4746 kernel_size_len: c_int,
4747 stride_data: *const i64,
4748 stride_len: c_int,
4749 padding_data: *const i64,
4750 padding_len: c_int,
4751 ceil_mode_: c_int,
4752 count_include_pad_: c_int,
4753 divisor_override_v: i64,
4754 divisor_override_null: i8,
4755 );
4756 pub fn atg_avg_pool2d_out(
4757 out__: *mut *mut C_tensor,
4758 out_: *mut C_tensor,
4759 self_: *mut C_tensor,
4760 kernel_size_data: *const i64,
4761 kernel_size_len: c_int,
4762 stride_data: *const i64,
4763 stride_len: c_int,
4764 padding_data: *const i64,
4765 padding_len: c_int,
4766 ceil_mode_: c_int,
4767 count_include_pad_: c_int,
4768 divisor_override_v: i64,
4769 divisor_override_null: i8,
4770 );
4771 pub fn atg_avg_pool3d(
4772 out__: *mut *mut C_tensor,
4773 self_: *mut C_tensor,
4774 kernel_size_data: *const i64,
4775 kernel_size_len: c_int,
4776 stride_data: *const i64,
4777 stride_len: c_int,
4778 padding_data: *const i64,
4779 padding_len: c_int,
4780 ceil_mode_: c_int,
4781 count_include_pad_: c_int,
4782 divisor_override_v: i64,
4783 divisor_override_null: i8,
4784 );
4785 pub fn atg_avg_pool3d_backward(
4786 out__: *mut *mut C_tensor,
4787 grad_output_: *mut C_tensor,
4788 self_: *mut C_tensor,
4789 kernel_size_data: *const i64,
4790 kernel_size_len: c_int,
4791 stride_data: *const i64,
4792 stride_len: c_int,
4793 padding_data: *const i64,
4794 padding_len: c_int,
4795 ceil_mode_: c_int,
4796 count_include_pad_: c_int,
4797 divisor_override_v: i64,
4798 divisor_override_null: i8,
4799 );
4800 pub fn atg_avg_pool3d_backward_grad_input(
4801 out__: *mut *mut C_tensor,
4802 grad_input_: *mut C_tensor,
4803 grad_output_: *mut C_tensor,
4804 self_: *mut C_tensor,
4805 kernel_size_data: *const i64,
4806 kernel_size_len: c_int,
4807 stride_data: *const i64,
4808 stride_len: c_int,
4809 padding_data: *const i64,
4810 padding_len: c_int,
4811 ceil_mode_: c_int,
4812 count_include_pad_: c_int,
4813 divisor_override_v: i64,
4814 divisor_override_null: i8,
4815 );
4816 pub fn atg_avg_pool3d_out(
4817 out__: *mut *mut C_tensor,
4818 out_: *mut C_tensor,
4819 self_: *mut C_tensor,
4820 kernel_size_data: *const i64,
4821 kernel_size_len: c_int,
4822 stride_data: *const i64,
4823 stride_len: c_int,
4824 padding_data: *const i64,
4825 padding_len: c_int,
4826 ceil_mode_: c_int,
4827 count_include_pad_: c_int,
4828 divisor_override_v: i64,
4829 divisor_override_null: i8,
4830 );
4831 pub fn atg_baddbmm(
4832 out__: *mut *mut C_tensor,
4833 self_: *mut C_tensor,
4834 batch1_: *mut C_tensor,
4835 batch2_: *mut C_tensor,
4836 beta_: *mut C_scalar,
4837 alpha_: *mut C_scalar,
4838 );
4839 pub fn atg_baddbmm_(
4840 out__: *mut *mut C_tensor,
4841 self_: *mut C_tensor,
4842 batch1_: *mut C_tensor,
4843 batch2_: *mut C_tensor,
4844 );
4845 pub fn atg_baddbmm_dtype(
4846 out__: *mut *mut C_tensor,
4847 self_: *mut C_tensor,
4848 batch1_: *mut C_tensor,
4849 batch2_: *mut C_tensor,
4850 out_dtype_: c_int,
4851 beta_: *mut C_scalar,
4852 alpha_: *mut C_scalar,
4853 );
4854 pub fn atg_baddbmm_dtype_out(
4855 out__: *mut *mut C_tensor,
4856 out_: *mut C_tensor,
4857 self_: *mut C_tensor,
4858 batch1_: *mut C_tensor,
4859 batch2_: *mut C_tensor,
4860 out_dtype_: c_int,
4861 );
4862 pub fn atg_baddbmm_out(
4863 out__: *mut *mut C_tensor,
4864 out_: *mut C_tensor,
4865 self_: *mut C_tensor,
4866 batch1_: *mut C_tensor,
4867 batch2_: *mut C_tensor,
4868 );
4869 pub fn atg_bartlett_window(
4870 out__: *mut *mut C_tensor,
4871 window_length_: i64,
4872 options_kind: c_int,
4873 options_device: c_int,
4874 );
4875 pub fn atg_bartlett_window_out(
4876 out__: *mut *mut C_tensor,
4877 out_: *mut C_tensor,
4878 window_length_: i64,
4879 );
4880 pub fn atg_bartlett_window_periodic(
4881 out__: *mut *mut C_tensor,
4882 window_length_: i64,
4883 periodic_: c_int,
4884 options_kind: c_int,
4885 options_device: c_int,
4886 );
4887 pub fn atg_bartlett_window_periodic_out(
4888 out__: *mut *mut C_tensor,
4889 out_: *mut C_tensor,
4890 window_length_: i64,
4891 periodic_: c_int,
4892 );
4893 pub fn atg_batch_norm(
4894 out__: *mut *mut C_tensor,
4895 input_: *mut C_tensor,
4896 weight_: *mut C_tensor,
4897 bias_: *mut C_tensor,
4898 running_mean_: *mut C_tensor,
4899 running_var_: *mut C_tensor,
4900 training_: c_int,
4901 momentum_: f64,
4902 eps_: f64,
4903 cudnn_enabled_: c_int,
4904 );
4905 pub fn atg_batch_norm_backward_elemt(
4906 out__: *mut *mut C_tensor,
4907 grad_out_: *mut C_tensor,
4908 input_: *mut C_tensor,
4909 mean_: *mut C_tensor,
4910 invstd_: *mut C_tensor,
4911 weight_: *mut C_tensor,
4912 sum_dy_: *mut C_tensor,
4913 sum_dy_xmu_: *mut C_tensor,
4914 count_: *mut C_tensor,
4915 );
4916 pub fn atg_batch_norm_backward_elemt_out(
4917 out__: *mut *mut C_tensor,
4918 out_: *mut C_tensor,
4919 grad_out_: *mut C_tensor,
4920 input_: *mut C_tensor,
4921 mean_: *mut C_tensor,
4922 invstd_: *mut C_tensor,
4923 weight_: *mut C_tensor,
4924 sum_dy_: *mut C_tensor,
4925 sum_dy_xmu_: *mut C_tensor,
4926 count_: *mut C_tensor,
4927 );
4928 pub fn atg_batch_norm_backward_reduce(
4929 out__: *mut *mut C_tensor,
4930 grad_out_: *mut C_tensor,
4931 input_: *mut C_tensor,
4932 mean_: *mut C_tensor,
4933 invstd_: *mut C_tensor,
4934 weight_: *mut C_tensor,
4935 input_g_: c_int,
4936 weight_g_: c_int,
4937 bias_g_: c_int,
4938 );
4939 pub fn atg_batch_norm_backward_reduce_out(
4940 out__: *mut *mut C_tensor,
4941 out0_: *mut C_tensor,
4942 out1_: *mut C_tensor,
4943 out2_: *mut C_tensor,
4944 out3_: *mut C_tensor,
4945 grad_out_: *mut C_tensor,
4946 input_: *mut C_tensor,
4947 mean_: *mut C_tensor,
4948 invstd_: *mut C_tensor,
4949 weight_: *mut C_tensor,
4950 input_g_: c_int,
4951 weight_g_: c_int,
4952 bias_g_: c_int,
4953 );
4954 pub fn atg_batch_norm_elemt(
4955 out__: *mut *mut C_tensor,
4956 input_: *mut C_tensor,
4957 weight_: *mut C_tensor,
4958 bias_: *mut C_tensor,
4959 mean_: *mut C_tensor,
4960 invstd_: *mut C_tensor,
4961 eps_: f64,
4962 );
4963 pub fn atg_batch_norm_elemt_out(
4964 out__: *mut *mut C_tensor,
4965 out_: *mut C_tensor,
4966 input_: *mut C_tensor,
4967 weight_: *mut C_tensor,
4968 bias_: *mut C_tensor,
4969 mean_: *mut C_tensor,
4970 invstd_: *mut C_tensor,
4971 eps_: f64,
4972 );
4973 pub fn atg_batch_norm_gather_stats(
4974 out__: *mut *mut C_tensor,
4975 input_: *mut C_tensor,
4976 mean_: *mut C_tensor,
4977 invstd_: *mut C_tensor,
4978 running_mean_: *mut C_tensor,
4979 running_var_: *mut C_tensor,
4980 momentum_: f64,
4981 eps_: f64,
4982 count_: i64,
4983 );
4984 pub fn atg_batch_norm_gather_stats_out(
4985 out__: *mut *mut C_tensor,
4986 out0_: *mut C_tensor,
4987 out1_: *mut C_tensor,
4988 input_: *mut C_tensor,
4989 mean_: *mut C_tensor,
4990 invstd_: *mut C_tensor,
4991 running_mean_: *mut C_tensor,
4992 running_var_: *mut C_tensor,
4993 momentum_: f64,
4994 eps_: f64,
4995 count_: i64,
4996 );
4997 pub fn atg_batch_norm_gather_stats_with_counts(
4998 out__: *mut *mut C_tensor,
4999 input_: *mut C_tensor,
5000 mean_: *mut C_tensor,
5001 invstd_: *mut C_tensor,
5002 running_mean_: *mut C_tensor,
5003 running_var_: *mut C_tensor,
5004 momentum_: f64,
5005 eps_: f64,
5006 counts_: *mut C_tensor,
5007 );
5008 pub fn atg_batch_norm_gather_stats_with_counts_out(
5009 out__: *mut *mut C_tensor,
5010 out0_: *mut C_tensor,
5011 out1_: *mut C_tensor,
5012 input_: *mut C_tensor,
5013 mean_: *mut C_tensor,
5014 invstd_: *mut C_tensor,
5015 running_mean_: *mut C_tensor,
5016 running_var_: *mut C_tensor,
5017 momentum_: f64,
5018 eps_: f64,
5019 counts_: *mut C_tensor,
5020 );
5021 pub fn atg_batch_norm_stats(out__: *mut *mut C_tensor, input_: *mut C_tensor, eps_: f64);
5022 pub fn atg_batch_norm_stats_out(
5023 out__: *mut *mut C_tensor,
5024 out0_: *mut C_tensor,
5025 out1_: *mut C_tensor,
5026 input_: *mut C_tensor,
5027 eps_: f64,
5028 );
5029 pub fn atg_batch_norm_update_stats(
5030 out__: *mut *mut C_tensor,
5031 input_: *mut C_tensor,
5032 running_mean_: *mut C_tensor,
5033 running_var_: *mut C_tensor,
5034 momentum_: f64,
5035 );
5036 pub fn atg_batch_norm_update_stats_out(
5037 out__: *mut *mut C_tensor,
5038 out0_: *mut C_tensor,
5039 out1_: *mut C_tensor,
5040 input_: *mut C_tensor,
5041 running_mean_: *mut C_tensor,
5042 running_var_: *mut C_tensor,
5043 momentum_: f64,
5044 );
5045 pub fn atg_bernoulli(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5046 pub fn atg_bernoulli_(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: *mut C_tensor);
5047 pub fn atg_bernoulli_float_(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: f64);
5048 pub fn atg_bernoulli_p(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: f64);
5049 pub fn atg_bernoulli_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: *mut C_tensor);
5050 pub fn atg_bilinear(
5051 out__: *mut *mut C_tensor,
5052 input1_: *mut C_tensor,
5053 input2_: *mut C_tensor,
5054 weight_: *mut C_tensor,
5055 bias_: *mut C_tensor,
5056 );
5057 pub fn atg_binary_cross_entropy(
5058 out__: *mut *mut C_tensor,
5059 self_: *mut C_tensor,
5060 target_: *mut C_tensor,
5061 weight_: *mut C_tensor,
5062 reduction_: i64,
5063 );
5064 pub fn atg_binary_cross_entropy_backward(
5065 out__: *mut *mut C_tensor,
5066 grad_output_: *mut C_tensor,
5067 self_: *mut C_tensor,
5068 target_: *mut C_tensor,
5069 weight_: *mut C_tensor,
5070 reduction_: i64,
5071 );
5072 pub fn atg_binary_cross_entropy_backward_grad_input(
5073 out__: *mut *mut C_tensor,
5074 grad_input_: *mut C_tensor,
5075 grad_output_: *mut C_tensor,
5076 self_: *mut C_tensor,
5077 target_: *mut C_tensor,
5078 weight_: *mut C_tensor,
5079 reduction_: i64,
5080 );
5081 pub fn atg_binary_cross_entropy_out(
5082 out__: *mut *mut C_tensor,
5083 out_: *mut C_tensor,
5084 self_: *mut C_tensor,
5085 target_: *mut C_tensor,
5086 weight_: *mut C_tensor,
5087 reduction_: i64,
5088 );
5089 pub fn atg_binary_cross_entropy_with_logits(
5090 out__: *mut *mut C_tensor,
5091 self_: *mut C_tensor,
5092 target_: *mut C_tensor,
5093 weight_: *mut C_tensor,
5094 pos_weight_: *mut C_tensor,
5095 reduction_: i64,
5096 );
5097 pub fn atg_binary_cross_entropy_with_logits_out(
5098 out__: *mut *mut C_tensor,
5099 out_: *mut C_tensor,
5100 self_: *mut C_tensor,
5101 target_: *mut C_tensor,
5102 weight_: *mut C_tensor,
5103 pos_weight_: *mut C_tensor,
5104 reduction_: i64,
5105 );
5106 pub fn atg_bincount(
5107 out__: *mut *mut C_tensor,
5108 self_: *mut C_tensor,
5109 weights_: *mut C_tensor,
5110 minlength_: i64,
5111 );
5112 pub fn atg_bincount_out(
5113 out__: *mut *mut C_tensor,
5114 out_: *mut C_tensor,
5115 self_: *mut C_tensor,
5116 weights_: *mut C_tensor,
5117 minlength_: i64,
5118 );
5119 pub fn atg_binomial(out__: *mut *mut C_tensor, count_: *mut C_tensor, prob_: *mut C_tensor);
5120 pub fn atg_binomial_out(
5121 out__: *mut *mut C_tensor,
5122 out_: *mut C_tensor,
5123 count_: *mut C_tensor,
5124 prob_: *mut C_tensor,
5125 );
5126 pub fn atg_bitwise_and(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
5127 pub fn atg_bitwise_and_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
5128 pub fn atg_bitwise_and_scalar_out(
5129 out__: *mut *mut C_tensor,
5130 out_: *mut C_tensor,
5131 self_: *mut C_tensor,
5132 other_: *mut C_scalar,
5133 );
5134 pub fn atg_bitwise_and_scalar_tensor(
5135 out__: *mut *mut C_tensor,
5136 self_scalar_: *mut C_scalar,
5137 other_: *mut C_tensor,
5138 );
5139 pub fn atg_bitwise_and_scalar_tensor_out(
5140 out__: *mut *mut C_tensor,
5141 out_: *mut C_tensor,
5142 self_scalar_: *mut C_scalar,
5143 other_: *mut C_tensor,
5144 );
5145 pub fn atg_bitwise_and_tensor(
5146 out__: *mut *mut C_tensor,
5147 self_: *mut C_tensor,
5148 other_: *mut C_tensor,
5149 );
5150 pub fn atg_bitwise_and_tensor_(
5151 out__: *mut *mut C_tensor,
5152 self_: *mut C_tensor,
5153 other_: *mut C_tensor,
5154 );
5155 pub fn atg_bitwise_and_tensor_out(
5156 out__: *mut *mut C_tensor,
5157 out_: *mut C_tensor,
5158 self_: *mut C_tensor,
5159 other_: *mut C_tensor,
5160 );
5161 pub fn atg_bitwise_left_shift(
5162 out__: *mut *mut C_tensor,
5163 self_: *mut C_tensor,
5164 other_: *mut C_tensor,
5165 );
5166 pub fn atg_bitwise_left_shift_(
5167 out__: *mut *mut C_tensor,
5168 self_: *mut C_tensor,
5169 other_: *mut C_tensor,
5170 );
5171 pub fn atg_bitwise_left_shift_scalar_tensor(
5172 out__: *mut *mut C_tensor,
5173 self_scalar_: *mut C_scalar,
5174 other_: *mut C_tensor,
5175 );
5176 pub fn atg_bitwise_left_shift_scalar_tensor_out(
5177 out__: *mut *mut C_tensor,
5178 out_: *mut C_tensor,
5179 self_scalar_: *mut C_scalar,
5180 other_: *mut C_tensor,
5181 );
5182 pub fn atg_bitwise_left_shift_tensor_out(
5183 out__: *mut *mut C_tensor,
5184 out_: *mut C_tensor,
5185 self_: *mut C_tensor,
5186 other_: *mut C_tensor,
5187 );
5188 pub fn atg_bitwise_left_shift_tensor_scalar(
5189 out__: *mut *mut C_tensor,
5190 self_: *mut C_tensor,
5191 other_: *mut C_scalar,
5192 );
5193 pub fn atg_bitwise_left_shift_tensor_scalar_(
5194 out__: *mut *mut C_tensor,
5195 self_: *mut C_tensor,
5196 other_: *mut C_scalar,
5197 );
5198 pub fn atg_bitwise_left_shift_tensor_scalar_out(
5199 out__: *mut *mut C_tensor,
5200 out_: *mut C_tensor,
5201 self_: *mut C_tensor,
5202 other_: *mut C_scalar,
5203 );
5204 pub fn atg_bitwise_not(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5205 pub fn atg_bitwise_not_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5206 pub fn atg_bitwise_not_out(
5207 out__: *mut *mut C_tensor,
5208 out_: *mut C_tensor,
5209 self_: *mut C_tensor,
5210 );
5211 pub fn atg_bitwise_or(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
5212 pub fn atg_bitwise_or_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
5213 pub fn atg_bitwise_or_scalar_out(
5214 out__: *mut *mut C_tensor,
5215 out_: *mut C_tensor,
5216 self_: *mut C_tensor,
5217 other_: *mut C_scalar,
5218 );
5219 pub fn atg_bitwise_or_scalar_tensor(
5220 out__: *mut *mut C_tensor,
5221 self_scalar_: *mut C_scalar,
5222 other_: *mut C_tensor,
5223 );
5224 pub fn atg_bitwise_or_scalar_tensor_out(
5225 out__: *mut *mut C_tensor,
5226 out_: *mut C_tensor,
5227 self_scalar_: *mut C_scalar,
5228 other_: *mut C_tensor,
5229 );
5230 pub fn atg_bitwise_or_tensor(
5231 out__: *mut *mut C_tensor,
5232 self_: *mut C_tensor,
5233 other_: *mut C_tensor,
5234 );
5235 pub fn atg_bitwise_or_tensor_(
5236 out__: *mut *mut C_tensor,
5237 self_: *mut C_tensor,
5238 other_: *mut C_tensor,
5239 );
5240 pub fn atg_bitwise_or_tensor_out(
5241 out__: *mut *mut C_tensor,
5242 out_: *mut C_tensor,
5243 self_: *mut C_tensor,
5244 other_: *mut C_tensor,
5245 );
5246 pub fn atg_bitwise_right_shift(
5247 out__: *mut *mut C_tensor,
5248 self_: *mut C_tensor,
5249 other_: *mut C_tensor,
5250 );
5251 pub fn atg_bitwise_right_shift_(
5252 out__: *mut *mut C_tensor,
5253 self_: *mut C_tensor,
5254 other_: *mut C_tensor,
5255 );
5256 pub fn atg_bitwise_right_shift_scalar_tensor(
5257 out__: *mut *mut C_tensor,
5258 self_scalar_: *mut C_scalar,
5259 other_: *mut C_tensor,
5260 );
5261 pub fn atg_bitwise_right_shift_scalar_tensor_out(
5262 out__: *mut *mut C_tensor,
5263 out_: *mut C_tensor,
5264 self_scalar_: *mut C_scalar,
5265 other_: *mut C_tensor,
5266 );
5267 pub fn atg_bitwise_right_shift_tensor_out(
5268 out__: *mut *mut C_tensor,
5269 out_: *mut C_tensor,
5270 self_: *mut C_tensor,
5271 other_: *mut C_tensor,
5272 );
5273 pub fn atg_bitwise_right_shift_tensor_scalar(
5274 out__: *mut *mut C_tensor,
5275 self_: *mut C_tensor,
5276 other_: *mut C_scalar,
5277 );
5278 pub fn atg_bitwise_right_shift_tensor_scalar_(
5279 out__: *mut *mut C_tensor,
5280 self_: *mut C_tensor,
5281 other_: *mut C_scalar,
5282 );
5283 pub fn atg_bitwise_right_shift_tensor_scalar_out(
5284 out__: *mut *mut C_tensor,
5285 out_: *mut C_tensor,
5286 self_: *mut C_tensor,
5287 other_: *mut C_scalar,
5288 );
5289 pub fn atg_bitwise_xor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
5290 pub fn atg_bitwise_xor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
5291 pub fn atg_bitwise_xor_scalar_out(
5292 out__: *mut *mut C_tensor,
5293 out_: *mut C_tensor,
5294 self_: *mut C_tensor,
5295 other_: *mut C_scalar,
5296 );
5297 pub fn atg_bitwise_xor_scalar_tensor(
5298 out__: *mut *mut C_tensor,
5299 self_scalar_: *mut C_scalar,
5300 other_: *mut C_tensor,
5301 );
5302 pub fn atg_bitwise_xor_scalar_tensor_out(
5303 out__: *mut *mut C_tensor,
5304 out_: *mut C_tensor,
5305 self_scalar_: *mut C_scalar,
5306 other_: *mut C_tensor,
5307 );
5308 pub fn atg_bitwise_xor_tensor(
5309 out__: *mut *mut C_tensor,
5310 self_: *mut C_tensor,
5311 other_: *mut C_tensor,
5312 );
5313 pub fn atg_bitwise_xor_tensor_(
5314 out__: *mut *mut C_tensor,
5315 self_: *mut C_tensor,
5316 other_: *mut C_tensor,
5317 );
5318 pub fn atg_bitwise_xor_tensor_out(
5319 out__: *mut *mut C_tensor,
5320 out_: *mut C_tensor,
5321 self_: *mut C_tensor,
5322 other_: *mut C_tensor,
5323 );
5324 pub fn atg_blackman_window(
5325 out__: *mut *mut C_tensor,
5326 window_length_: i64,
5327 options_kind: c_int,
5328 options_device: c_int,
5329 );
5330 pub fn atg_blackman_window_out(
5331 out__: *mut *mut C_tensor,
5332 out_: *mut C_tensor,
5333 window_length_: i64,
5334 );
5335 pub fn atg_blackman_window_periodic(
5336 out__: *mut *mut C_tensor,
5337 window_length_: i64,
5338 periodic_: c_int,
5339 options_kind: c_int,
5340 options_device: c_int,
5341 );
5342 pub fn atg_blackman_window_periodic_out(
5343 out__: *mut *mut C_tensor,
5344 out_: *mut C_tensor,
5345 window_length_: i64,
5346 periodic_: c_int,
5347 );
5348 pub fn atg_block_diag(
5349 out__: *mut *mut C_tensor,
5350 tensors_data: *const *mut C_tensor,
5351 tensors_len: c_int,
5352 );
5353 pub fn atg_block_diag_out(
5354 out__: *mut *mut C_tensor,
5355 out_: *mut C_tensor,
5356 tensors_data: *const *mut C_tensor,
5357 tensors_len: c_int,
5358 );
5359 pub fn atg_bmm(out__: *mut *mut C_tensor, self_: *mut C_tensor, mat2_: *mut C_tensor);
5360 pub fn atg_bmm_dtype(
5361 out__: *mut *mut C_tensor,
5362 self_: *mut C_tensor,
5363 mat2_: *mut C_tensor,
5364 out_dtype_: c_int,
5365 );
5366 pub fn atg_bmm_dtype_out(
5367 out__: *mut *mut C_tensor,
5368 out_: *mut C_tensor,
5369 self_: *mut C_tensor,
5370 mat2_: *mut C_tensor,
5371 out_dtype_: c_int,
5372 );
5373 pub fn atg_bmm_out(
5374 out__: *mut *mut C_tensor,
5375 out_: *mut C_tensor,
5376 self_: *mut C_tensor,
5377 mat2_: *mut C_tensor,
5378 );
5379 pub fn atg_broadcast_tensors(
5380 tensors_data: *const *mut C_tensor,
5381 tensors_len: c_int,
5382 ) -> *mut *mut C_tensor;
5383 pub fn atg_broadcast_to(
5384 out__: *mut *mut C_tensor,
5385 self_: *mut C_tensor,
5386 size_data: *const i64,
5387 size_len: c_int,
5388 );
5389 pub fn atg_bucketize(
5390 out__: *mut *mut C_tensor,
5391 self_: *mut C_tensor,
5392 boundaries_: *mut C_tensor,
5393 out_int32_: c_int,
5394 right_: c_int,
5395 );
5396 pub fn atg_bucketize_scalar(
5397 out__: *mut *mut C_tensor,
5398 self_scalar_: *mut C_scalar,
5399 boundaries_: *mut C_tensor,
5400 out_int32_: c_int,
5401 right_: c_int,
5402 );
5403 pub fn atg_bucketize_scalar_out(
5404 out__: *mut *mut C_tensor,
5405 out_: *mut C_tensor,
5406 self_scalar_: *mut C_scalar,
5407 boundaries_: *mut C_tensor,
5408 out_int32_: c_int,
5409 right_: c_int,
5410 );
5411 pub fn atg_bucketize_tensor_out(
5412 out__: *mut *mut C_tensor,
5413 out_: *mut C_tensor,
5414 self_: *mut C_tensor,
5415 boundaries_: *mut C_tensor,
5416 out_int32_: c_int,
5417 right_: c_int,
5418 );
5419 pub fn atg_can_cast(from__: c_int, to_: c_int) -> c_int;
5420 pub fn atg_cartesian_prod(
5421 out__: *mut *mut C_tensor,
5422 tensors_data: *const *mut C_tensor,
5423 tensors_len: c_int,
5424 );
5425 pub fn atg_cat(
5426 out__: *mut *mut C_tensor,
5427 tensors_data: *const *mut C_tensor,
5428 tensors_len: c_int,
5429 dim_: i64,
5430 );
5431 pub fn atg_cat_out(
5432 out__: *mut *mut C_tensor,
5433 out_: *mut C_tensor,
5434 tensors_data: *const *mut C_tensor,
5435 tensors_len: c_int,
5436 dim_: i64,
5437 );
5438 pub fn atg_cauchy(out__: *mut *mut C_tensor, self_: *mut C_tensor, median_: f64, sigma_: f64);
5439 pub fn atg_cauchy_(out__: *mut *mut C_tensor, self_: *mut C_tensor, median_: f64, sigma_: f64);
5440 pub fn atg_cauchy_out(
5441 out__: *mut *mut C_tensor,
5442 out_: *mut C_tensor,
5443 self_: *mut C_tensor,
5444 median_: f64,
5445 sigma_: f64,
5446 );
5447 pub fn atg_ccol_indices(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5448 pub fn atg_ccol_indices_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5449 pub fn atg_ccol_indices_copy_out(
5450 out__: *mut *mut C_tensor,
5451 out_: *mut C_tensor,
5452 self_: *mut C_tensor,
5453 );
5454 pub fn atg_cdist(
5455 out__: *mut *mut C_tensor,
5456 x1_: *mut C_tensor,
5457 x2_: *mut C_tensor,
5458 p_: f64,
5459 compute_mode_v: i64,
5460 compute_mode_null: i8,
5461 );
5462 pub fn atg_ceil(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5463 pub fn atg_ceil_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5464 pub fn atg_ceil_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
5465 pub fn atg_celu(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5466 pub fn atg_celu_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5467 pub fn atg_celu_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
5468 pub fn atg_chain_matmul(
5469 out__: *mut *mut C_tensor,
5470 matrices_data: *const *mut C_tensor,
5471 matrices_len: c_int,
5472 );
5473 pub fn atg_chain_matmul_out(
5474 out__: *mut *mut C_tensor,
5475 out_: *mut C_tensor,
5476 matrices_data: *const *mut C_tensor,
5477 matrices_len: c_int,
5478 );
5479 pub fn atg_chalf(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5480 pub fn atg_channel_shuffle(out__: *mut *mut C_tensor, self_: *mut C_tensor, groups_: i64);
5481 pub fn atg_channel_shuffle_out(
5482 out__: *mut *mut C_tensor,
5483 out_: *mut C_tensor,
5484 self_: *mut C_tensor,
5485 groups_: i64,
5486 );
5487 pub fn atg_cholesky(out__: *mut *mut C_tensor, self_: *mut C_tensor, upper_: c_int);
5488 pub fn atg_cholesky_inverse(out__: *mut *mut C_tensor, self_: *mut C_tensor, upper_: c_int);
5489 pub fn atg_cholesky_inverse_out(
5490 out__: *mut *mut C_tensor,
5491 out_: *mut C_tensor,
5492 self_: *mut C_tensor,
5493 upper_: c_int,
5494 );
5495 pub fn atg_cholesky_out(
5496 out__: *mut *mut C_tensor,
5497 out_: *mut C_tensor,
5498 self_: *mut C_tensor,
5499 upper_: c_int,
5500 );
5501 pub fn atg_cholesky_solve(
5502 out__: *mut *mut C_tensor,
5503 self_: *mut C_tensor,
5504 input2_: *mut C_tensor,
5505 upper_: c_int,
5506 );
5507 pub fn atg_cholesky_solve_out(
5508 out__: *mut *mut C_tensor,
5509 out_: *mut C_tensor,
5510 self_: *mut C_tensor,
5511 input2_: *mut C_tensor,
5512 upper_: c_int,
5513 );
5514 pub fn atg_choose_qparams_optimized(
5515 out__: *mut *mut C_tensor,
5516 input_: *mut C_tensor,
5517 numel_: i64,
5518 n_bins_: i64,
5519 ratio_: f64,
5520 bit_width_: i64,
5521 );
5522 pub fn atg_chunk(self_: *mut C_tensor, chunks_: i64, dim_: i64) -> *mut *mut C_tensor;
5523 pub fn atg_clamp(
5524 out__: *mut *mut C_tensor,
5525 self_: *mut C_tensor,
5526 min_: *mut C_scalar,
5527 max_: *mut C_scalar,
5528 );
5529 pub fn atg_clamp_(
5530 out__: *mut *mut C_tensor,
5531 self_: *mut C_tensor,
5532 min_: *mut C_scalar,
5533 max_: *mut C_scalar,
5534 );
5535 pub fn atg_clamp_max(out__: *mut *mut C_tensor, self_: *mut C_tensor, max_: *mut C_scalar);
5536 pub fn atg_clamp_max_(out__: *mut *mut C_tensor, self_: *mut C_tensor, max_: *mut C_scalar);
5537 pub fn atg_clamp_max_out(
5538 out__: *mut *mut C_tensor,
5539 out_: *mut C_tensor,
5540 self_: *mut C_tensor,
5541 max_: *mut C_scalar,
5542 );
5543 pub fn atg_clamp_max_tensor(
5544 out__: *mut *mut C_tensor,
5545 self_: *mut C_tensor,
5546 max_: *mut C_tensor,
5547 );
5548 pub fn atg_clamp_max_tensor_(
5549 out__: *mut *mut C_tensor,
5550 self_: *mut C_tensor,
5551 max_: *mut C_tensor,
5552 );
5553 pub fn atg_clamp_max_tensor_out(
5554 out__: *mut *mut C_tensor,
5555 out_: *mut C_tensor,
5556 self_: *mut C_tensor,
5557 max_: *mut C_tensor,
5558 );
5559 pub fn atg_clamp_min(out__: *mut *mut C_tensor, self_: *mut C_tensor, min_: *mut C_scalar);
5560 pub fn atg_clamp_min_(out__: *mut *mut C_tensor, self_: *mut C_tensor, min_: *mut C_scalar);
5561 pub fn atg_clamp_min_out(
5562 out__: *mut *mut C_tensor,
5563 out_: *mut C_tensor,
5564 self_: *mut C_tensor,
5565 min_: *mut C_scalar,
5566 );
5567 pub fn atg_clamp_min_tensor(
5568 out__: *mut *mut C_tensor,
5569 self_: *mut C_tensor,
5570 min_: *mut C_tensor,
5571 );
5572 pub fn atg_clamp_min_tensor_(
5573 out__: *mut *mut C_tensor,
5574 self_: *mut C_tensor,
5575 min_: *mut C_tensor,
5576 );
5577 pub fn atg_clamp_min_tensor_out(
5578 out__: *mut *mut C_tensor,
5579 out_: *mut C_tensor,
5580 self_: *mut C_tensor,
5581 min_: *mut C_tensor,
5582 );
5583 pub fn atg_clamp_out(
5584 out__: *mut *mut C_tensor,
5585 out_: *mut C_tensor,
5586 self_: *mut C_tensor,
5587 min_: *mut C_scalar,
5588 max_: *mut C_scalar,
5589 );
5590 pub fn atg_clamp_tensor(
5591 out__: *mut *mut C_tensor,
5592 self_: *mut C_tensor,
5593 min_: *mut C_tensor,
5594 max_: *mut C_tensor,
5595 );
5596 pub fn atg_clamp_tensor_(
5597 out__: *mut *mut C_tensor,
5598 self_: *mut C_tensor,
5599 min_: *mut C_tensor,
5600 max_: *mut C_tensor,
5601 );
5602 pub fn atg_clamp_tensor_out(
5603 out__: *mut *mut C_tensor,
5604 out_: *mut C_tensor,
5605 self_: *mut C_tensor,
5606 min_: *mut C_tensor,
5607 max_: *mut C_tensor,
5608 );
5609 pub fn atg_clip(
5610 out__: *mut *mut C_tensor,
5611 self_: *mut C_tensor,
5612 min_: *mut C_scalar,
5613 max_: *mut C_scalar,
5614 );
5615 pub fn atg_clip_(
5616 out__: *mut *mut C_tensor,
5617 self_: *mut C_tensor,
5618 min_: *mut C_scalar,
5619 max_: *mut C_scalar,
5620 );
5621 pub fn atg_clip_out(
5622 out__: *mut *mut C_tensor,
5623 out_: *mut C_tensor,
5624 self_: *mut C_tensor,
5625 min_: *mut C_scalar,
5626 max_: *mut C_scalar,
5627 );
5628 pub fn atg_clip_tensor(
5629 out__: *mut *mut C_tensor,
5630 self_: *mut C_tensor,
5631 min_: *mut C_tensor,
5632 max_: *mut C_tensor,
5633 );
5634 pub fn atg_clip_tensor_(
5635 out__: *mut *mut C_tensor,
5636 self_: *mut C_tensor,
5637 min_: *mut C_tensor,
5638 max_: *mut C_tensor,
5639 );
5640 pub fn atg_clip_tensor_out(
5641 out__: *mut *mut C_tensor,
5642 out_: *mut C_tensor,
5643 self_: *mut C_tensor,
5644 min_: *mut C_tensor,
5645 max_: *mut C_tensor,
5646 );
5647 pub fn atg_clone(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
5648 pub fn atg_coalesce(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5649 pub fn atg_col2im(
5650 out__: *mut *mut C_tensor,
5651 self_: *mut C_tensor,
5652 output_size_data: *const i64,
5653 output_size_len: c_int,
5654 kernel_size_data: *const i64,
5655 kernel_size_len: c_int,
5656 dilation_data: *const i64,
5657 dilation_len: c_int,
5658 padding_data: *const i64,
5659 padding_len: c_int,
5660 stride_data: *const i64,
5661 stride_len: c_int,
5662 );
5663 pub fn atg_col2im_out(
5664 out__: *mut *mut C_tensor,
5665 out_: *mut C_tensor,
5666 self_: *mut C_tensor,
5667 output_size_data: *const i64,
5668 output_size_len: c_int,
5669 kernel_size_data: *const i64,
5670 kernel_size_len: c_int,
5671 dilation_data: *const i64,
5672 dilation_len: c_int,
5673 padding_data: *const i64,
5674 padding_len: c_int,
5675 stride_data: *const i64,
5676 stride_len: c_int,
5677 );
5678 pub fn atg_col_indices(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5679 pub fn atg_col_indices_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5680 pub fn atg_col_indices_copy_out(
5681 out__: *mut *mut C_tensor,
5682 out_: *mut C_tensor,
5683 self_: *mut C_tensor,
5684 );
5685 pub fn atg_column_stack(
5686 out__: *mut *mut C_tensor,
5687 tensors_data: *const *mut C_tensor,
5688 tensors_len: c_int,
5689 );
5690 pub fn atg_column_stack_out(
5691 out__: *mut *mut C_tensor,
5692 out_: *mut C_tensor,
5693 tensors_data: *const *mut C_tensor,
5694 tensors_len: c_int,
5695 );
5696 pub fn atg_combinations(
5697 out__: *mut *mut C_tensor,
5698 self_: *mut C_tensor,
5699 r_: i64,
5700 with_replacement_: c_int,
5701 );
5702 pub fn atg_complex(out__: *mut *mut C_tensor, real_: *mut C_tensor, imag_: *mut C_tensor);
5703 pub fn atg_complex_out(
5704 out__: *mut *mut C_tensor,
5705 out_: *mut C_tensor,
5706 real_: *mut C_tensor,
5707 imag_: *mut C_tensor,
5708 );
5709 pub fn atg_concat(
5710 out__: *mut *mut C_tensor,
5711 tensors_data: *const *mut C_tensor,
5712 tensors_len: c_int,
5713 dim_: i64,
5714 );
5715 pub fn atg_concat_out(
5716 out__: *mut *mut C_tensor,
5717 out_: *mut C_tensor,
5718 tensors_data: *const *mut C_tensor,
5719 tensors_len: c_int,
5720 dim_: i64,
5721 );
5722 pub fn atg_concatenate(
5723 out__: *mut *mut C_tensor,
5724 tensors_data: *const *mut C_tensor,
5725 tensors_len: c_int,
5726 dim_: i64,
5727 );
5728 pub fn atg_concatenate_out(
5729 out__: *mut *mut C_tensor,
5730 out_: *mut C_tensor,
5731 tensors_data: *const *mut C_tensor,
5732 tensors_len: c_int,
5733 dim_: i64,
5734 );
5735 pub fn atg_conj(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5736 pub fn atg_conj_physical(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5737 pub fn atg_conj_physical_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5738 pub fn atg_conj_physical_out(
5739 out__: *mut *mut C_tensor,
5740 out_: *mut C_tensor,
5741 self_: *mut C_tensor,
5742 );
5743 pub fn atg_constant_pad_nd(
5744 out__: *mut *mut C_tensor,
5745 self_: *mut C_tensor,
5746 pad_data: *const i64,
5747 pad_len: c_int,
5748 );
5749 pub fn atg_constant_pad_nd_out(
5750 out__: *mut *mut C_tensor,
5751 out_: *mut C_tensor,
5752 self_: *mut C_tensor,
5753 pad_data: *const i64,
5754 pad_len: c_int,
5755 );
5756 pub fn atg_contiguous(out__: *mut *mut C_tensor, self_: *mut C_tensor);
5757 pub fn atg_conv1d(
5758 out__: *mut *mut C_tensor,
5759 input_: *mut C_tensor,
5760 weight_: *mut C_tensor,
5761 bias_: *mut C_tensor,
5762 stride_data: *const i64,
5763 stride_len: c_int,
5764 padding_data: *const i64,
5765 padding_len: c_int,
5766 dilation_data: *const i64,
5767 dilation_len: c_int,
5768 groups_: i64,
5769 );
5770 pub fn atg_conv1d_padding(
5771 out__: *mut *mut C_tensor,
5772 input_: *mut C_tensor,
5773 weight_: *mut C_tensor,
5774 bias_: *mut C_tensor,
5775 stride_data: *const i64,
5776 stride_len: c_int,
5777 padding_ptr: *const u8,
5778 padding_len: c_int,
5779 dilation_data: *const i64,
5780 dilation_len: c_int,
5781 groups_: i64,
5782 );
5783 pub fn atg_conv2d(
5784 out__: *mut *mut C_tensor,
5785 input_: *mut C_tensor,
5786 weight_: *mut C_tensor,
5787 bias_: *mut C_tensor,
5788 stride_data: *const i64,
5789 stride_len: c_int,
5790 padding_data: *const i64,
5791 padding_len: c_int,
5792 dilation_data: *const i64,
5793 dilation_len: c_int,
5794 groups_: i64,
5795 );
5796 pub fn atg_conv2d_padding(
5797 out__: *mut *mut C_tensor,
5798 input_: *mut C_tensor,
5799 weight_: *mut C_tensor,
5800 bias_: *mut C_tensor,
5801 stride_data: *const i64,
5802 stride_len: c_int,
5803 padding_ptr: *const u8,
5804 padding_len: c_int,
5805 dilation_data: *const i64,
5806 dilation_len: c_int,
5807 groups_: i64,
5808 );
5809 pub fn atg_conv3d(
5810 out__: *mut *mut C_tensor,
5811 input_: *mut C_tensor,
5812 weight_: *mut C_tensor,
5813 bias_: *mut C_tensor,
5814 stride_data: *const i64,
5815 stride_len: c_int,
5816 padding_data: *const i64,
5817 padding_len: c_int,
5818 dilation_data: *const i64,
5819 dilation_len: c_int,
5820 groups_: i64,
5821 );
5822 pub fn atg_conv3d_padding(
5823 out__: *mut *mut C_tensor,
5824 input_: *mut C_tensor,
5825 weight_: *mut C_tensor,
5826 bias_: *mut C_tensor,
5827 stride_data: *const i64,
5828 stride_len: c_int,
5829 padding_ptr: *const u8,
5830 padding_len: c_int,
5831 dilation_data: *const i64,
5832 dilation_len: c_int,
5833 groups_: i64,
5834 );
5835 pub fn atg_conv_depthwise3d(
5836 out__: *mut *mut C_tensor,
5837 self_: *mut C_tensor,
5838 weight_: *mut C_tensor,
5839 kernel_size_data: *const i64,
5840 kernel_size_len: c_int,
5841 bias_: *mut C_tensor,
5842 stride_data: *const i64,
5843 stride_len: c_int,
5844 padding_data: *const i64,
5845 padding_len: c_int,
5846 dilation_data: *const i64,
5847 dilation_len: c_int,
5848 );
5849 pub fn atg_conv_depthwise3d_out(
5850 out__: *mut *mut C_tensor,
5851 out_: *mut C_tensor,
5852 self_: *mut C_tensor,
5853 weight_: *mut C_tensor,
5854 kernel_size_data: *const i64,
5855 kernel_size_len: c_int,
5856 bias_: *mut C_tensor,
5857 stride_data: *const i64,
5858 stride_len: c_int,
5859 padding_data: *const i64,
5860 padding_len: c_int,
5861 dilation_data: *const i64,
5862 dilation_len: c_int,
5863 );
5864 pub fn atg_conv_tbc(
5865 out__: *mut *mut C_tensor,
5866 self_: *mut C_tensor,
5867 weight_: *mut C_tensor,
5868 bias_: *mut C_tensor,
5869 pad_: i64,
5870 );
5871 pub fn atg_conv_tbc_backward(
5872 out__: *mut *mut C_tensor,
5873 self_: *mut C_tensor,
5874 input_: *mut C_tensor,
5875 weight_: *mut C_tensor,
5876 bias_: *mut C_tensor,
5877 pad_: i64,
5878 );
5879 pub fn atg_conv_tbc_out(
5880 out__: *mut *mut C_tensor,
5881 out_: *mut C_tensor,
5882 self_: *mut C_tensor,
5883 weight_: *mut C_tensor,
5884 bias_: *mut C_tensor,
5885 pad_: i64,
5886 );
5887 pub fn atg_conv_transpose1d(
5888 out__: *mut *mut C_tensor,
5889 input_: *mut C_tensor,
5890 weight_: *mut C_tensor,
5891 bias_: *mut C_tensor,
5892 stride_data: *const i64,
5893 stride_len: c_int,
5894 padding_data: *const i64,
5895 padding_len: c_int,
5896 output_padding_data: *const i64,
5897 output_padding_len: c_int,
5898 groups_: i64,
5899 dilation_data: *const i64,
5900 dilation_len: c_int,
5901 );
5902 pub fn atg_conv_transpose2d(
5903 out__: *mut *mut C_tensor,
5904 input_: *mut C_tensor,
5905 weight_: *mut C_tensor,
5906 bias_: *mut C_tensor,
5907 stride_data: *const i64,
5908 stride_len: c_int,
5909 padding_data: *const i64,
5910 padding_len: c_int,
5911 output_padding_data: *const i64,
5912 output_padding_len: c_int,
5913 groups_: i64,
5914 dilation_data: *const i64,
5915 dilation_len: c_int,
5916 );
5917 pub fn atg_conv_transpose3d(
5918 out__: *mut *mut C_tensor,
5919 input_: *mut C_tensor,
5920 weight_: *mut C_tensor,
5921 bias_: *mut C_tensor,
5922 stride_data: *const i64,
5923 stride_len: c_int,
5924 padding_data: *const i64,
5925 padding_len: c_int,
5926 output_padding_data: *const i64,
5927 output_padding_len: c_int,
5928 groups_: i64,
5929 dilation_data: *const i64,
5930 dilation_len: c_int,
5931 );
5932 pub fn atg_convolution(
5933 out__: *mut *mut C_tensor,
5934 input_: *mut C_tensor,
5935 weight_: *mut C_tensor,
5936 bias_: *mut C_tensor,
5937 stride_data: *const i64,
5938 stride_len: c_int,
5939 padding_data: *const i64,
5940 padding_len: c_int,
5941 dilation_data: *const i64,
5942 dilation_len: c_int,
5943 transposed_: c_int,
5944 output_padding_data: *const i64,
5945 output_padding_len: c_int,
5946 groups_: i64,
5947 );
5948 pub fn atg_convolution_out(
5949 out__: *mut *mut C_tensor,
5950 out_: *mut C_tensor,
5951 input_: *mut C_tensor,
5952 weight_: *mut C_tensor,
5953 bias_: *mut C_tensor,
5954 stride_data: *const i64,
5955 stride_len: c_int,
5956 padding_data: *const i64,
5957 padding_len: c_int,
5958 dilation_data: *const i64,
5959 dilation_len: c_int,
5960 transposed_: c_int,
5961 output_padding_data: *const i64,
5962 output_padding_len: c_int,
5963 groups_: i64,
5964 );
5965 pub fn atg_convolution_overrideable(
5966 out__: *mut *mut C_tensor,
5967 input_: *mut C_tensor,
5968 weight_: *mut C_tensor,
5969 bias_: *mut C_tensor,
5970 stride_data: *const i64,
5971 stride_len: c_int,
5972 padding_data: *const i64,
5973 padding_len: c_int,
5974 dilation_data: *const i64,
5975 dilation_len: c_int,
5976 transposed_: c_int,
5977 output_padding_data: *const i64,
5978 output_padding_len: c_int,
5979 groups_: i64,
5980 );
5981 pub fn atg_convolution_overrideable_out(
5982 out__: *mut *mut C_tensor,
5983 out_: *mut C_tensor,
5984 input_: *mut C_tensor,
5985 weight_: *mut C_tensor,
5986 bias_: *mut C_tensor,
5987 stride_data: *const i64,
5988 stride_len: c_int,
5989 padding_data: *const i64,
5990 padding_len: c_int,
5991 dilation_data: *const i64,
5992 dilation_len: c_int,
5993 transposed_: c_int,
5994 output_padding_data: *const i64,
5995 output_padding_len: c_int,
5996 groups_: i64,
5997 );
5998 pub fn atg_copy_sparse_to_sparse(
5999 out__: *mut *mut C_tensor,
6000 self_: *mut C_tensor,
6001 src_: *mut C_tensor,
6002 non_blocking_: c_int,
6003 );
6004 pub fn atg_copy_sparse_to_sparse_(
6005 out__: *mut *mut C_tensor,
6006 self_: *mut C_tensor,
6007 src_: *mut C_tensor,
6008 non_blocking_: c_int,
6009 );
6010 pub fn atg_copy_sparse_to_sparse_out(
6011 out__: *mut *mut C_tensor,
6012 out_: *mut C_tensor,
6013 self_: *mut C_tensor,
6014 src_: *mut C_tensor,
6015 non_blocking_: c_int,
6016 );
6017 pub fn atg_copysign(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6018 pub fn atg_copysign_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6019 pub fn atg_copysign_out(
6020 out__: *mut *mut C_tensor,
6021 out_: *mut C_tensor,
6022 self_: *mut C_tensor,
6023 other_: *mut C_tensor,
6024 );
6025 pub fn atg_copysign_scalar(
6026 out__: *mut *mut C_tensor,
6027 self_: *mut C_tensor,
6028 other_: *mut C_scalar,
6029 );
6030 pub fn atg_copysign_scalar_(
6031 out__: *mut *mut C_tensor,
6032 self_: *mut C_tensor,
6033 other_: *mut C_scalar,
6034 );
6035 pub fn atg_copysign_scalar_out(
6036 out__: *mut *mut C_tensor,
6037 out_: *mut C_tensor,
6038 self_: *mut C_tensor,
6039 other_: *mut C_scalar,
6040 );
6041 pub fn atg_corrcoef(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6042 pub fn atg_cos(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6043 pub fn atg_cos_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6044 pub fn atg_cos_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6045 pub fn atg_cosh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6046 pub fn atg_cosh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6047 pub fn atg_cosh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6048 pub fn atg_cosine_embedding_loss(
6049 out__: *mut *mut C_tensor,
6050 input1_: *mut C_tensor,
6051 input2_: *mut C_tensor,
6052 target_: *mut C_tensor,
6053 margin_: f64,
6054 reduction_: i64,
6055 );
6056 pub fn atg_cosine_similarity(
6057 out__: *mut *mut C_tensor,
6058 x1_: *mut C_tensor,
6059 x2_: *mut C_tensor,
6060 dim_: i64,
6061 eps_: f64,
6062 );
6063 pub fn atg_count_nonzero(
6064 out__: *mut *mut C_tensor,
6065 self_: *mut C_tensor,
6066 dim_v: i64,
6067 dim_null: i8,
6068 );
6069 pub fn atg_count_nonzero_dim_intlist(
6070 out__: *mut *mut C_tensor,
6071 self_: *mut C_tensor,
6072 dim_data: *const i64,
6073 dim_len: c_int,
6074 );
6075 pub fn atg_count_nonzero_dim_intlist_out(
6076 out__: *mut *mut C_tensor,
6077 out_: *mut C_tensor,
6078 self_: *mut C_tensor,
6079 dim_data: *const i64,
6080 dim_len: c_int,
6081 );
6082 pub fn atg_count_nonzero_out(
6083 out__: *mut *mut C_tensor,
6084 out_: *mut C_tensor,
6085 self_: *mut C_tensor,
6086 dim_v: i64,
6087 dim_null: i8,
6088 );
6089 pub fn atg_cov(
6090 out__: *mut *mut C_tensor,
6091 self_: *mut C_tensor,
6092 correction_: i64,
6093 fweights_: *mut C_tensor,
6094 aweights_: *mut C_tensor,
6095 );
6096 pub fn atg_cross(
6097 out__: *mut *mut C_tensor,
6098 self_: *mut C_tensor,
6099 other_: *mut C_tensor,
6100 dim_v: i64,
6101 dim_null: i8,
6102 );
6103 pub fn atg_cross_entropy_loss(
6104 out__: *mut *mut C_tensor,
6105 self_: *mut C_tensor,
6106 target_: *mut C_tensor,
6107 weight_: *mut C_tensor,
6108 reduction_: i64,
6109 ignore_index_: i64,
6110 label_smoothing_: f64,
6111 );
6112 pub fn atg_cross_out(
6113 out__: *mut *mut C_tensor,
6114 out_: *mut C_tensor,
6115 self_: *mut C_tensor,
6116 other_: *mut C_tensor,
6117 dim_v: i64,
6118 dim_null: i8,
6119 );
6120 pub fn atg_crow_indices(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6121 pub fn atg_crow_indices_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6122 pub fn atg_crow_indices_copy_out(
6123 out__: *mut *mut C_tensor,
6124 out_: *mut C_tensor,
6125 self_: *mut C_tensor,
6126 );
6127 pub fn atg_ctc_loss(
6128 out__: *mut *mut C_tensor,
6129 log_probs_: *mut C_tensor,
6130 targets_: *mut C_tensor,
6131 input_lengths_data: *const i64,
6132 input_lengths_len: c_int,
6133 target_lengths_data: *const i64,
6134 target_lengths_len: c_int,
6135 blank_: i64,
6136 reduction_: i64,
6137 zero_infinity_: c_int,
6138 );
6139 pub fn atg_ctc_loss_tensor(
6140 out__: *mut *mut C_tensor,
6141 log_probs_: *mut C_tensor,
6142 targets_: *mut C_tensor,
6143 input_lengths_: *mut C_tensor,
6144 target_lengths_: *mut C_tensor,
6145 blank_: i64,
6146 reduction_: i64,
6147 zero_infinity_: c_int,
6148 );
6149 pub fn atg_cudnn_affine_grid_generator(
6150 out__: *mut *mut C_tensor,
6151 theta_: *mut C_tensor,
6152 n_: i64,
6153 C_: i64,
6154 H_: i64,
6155 W_: i64,
6156 );
6157 pub fn atg_cudnn_affine_grid_generator_backward(
6158 out__: *mut *mut C_tensor,
6159 grad_: *mut C_tensor,
6160 n_: i64,
6161 C_: i64,
6162 H_: i64,
6163 W_: i64,
6164 );
6165 pub fn atg_cudnn_affine_grid_generator_backward_out(
6166 out__: *mut *mut C_tensor,
6167 out_: *mut C_tensor,
6168 grad_: *mut C_tensor,
6169 n_: i64,
6170 C_: i64,
6171 H_: i64,
6172 W_: i64,
6173 );
6174 pub fn atg_cudnn_affine_grid_generator_out(
6175 out__: *mut *mut C_tensor,
6176 out_: *mut C_tensor,
6177 theta_: *mut C_tensor,
6178 n_: i64,
6179 C_: i64,
6180 H_: i64,
6181 W_: i64,
6182 );
6183 pub fn atg_cudnn_batch_norm(
6184 out__: *mut *mut C_tensor,
6185 input_: *mut C_tensor,
6186 weight_: *mut C_tensor,
6187 bias_: *mut C_tensor,
6188 running_mean_: *mut C_tensor,
6189 running_var_: *mut C_tensor,
6190 training_: c_int,
6191 exponential_average_factor_: f64,
6192 epsilon_: f64,
6193 );
6194 pub fn atg_cudnn_batch_norm_backward(
6195 out__: *mut *mut C_tensor,
6196 input_: *mut C_tensor,
6197 grad_output_: *mut C_tensor,
6198 weight_: *mut C_tensor,
6199 running_mean_: *mut C_tensor,
6200 running_var_: *mut C_tensor,
6201 save_mean_: *mut C_tensor,
6202 save_var_: *mut C_tensor,
6203 epsilon_: f64,
6204 reserveSpace_: *mut C_tensor,
6205 );
6206 pub fn atg_cudnn_batch_norm_backward_out(
6207 out__: *mut *mut C_tensor,
6208 out0_: *mut C_tensor,
6209 out1_: *mut C_tensor,
6210 out2_: *mut C_tensor,
6211 input_: *mut C_tensor,
6212 grad_output_: *mut C_tensor,
6213 weight_: *mut C_tensor,
6214 running_mean_: *mut C_tensor,
6215 running_var_: *mut C_tensor,
6216 save_mean_: *mut C_tensor,
6217 save_var_: *mut C_tensor,
6218 epsilon_: f64,
6219 reserveSpace_: *mut C_tensor,
6220 );
6221 pub fn atg_cudnn_batch_norm_out(
6222 out__: *mut *mut C_tensor,
6223 out0_: *mut C_tensor,
6224 out1_: *mut C_tensor,
6225 out2_: *mut C_tensor,
6226 out3_: *mut C_tensor,
6227 input_: *mut C_tensor,
6228 weight_: *mut C_tensor,
6229 bias_: *mut C_tensor,
6230 running_mean_: *mut C_tensor,
6231 running_var_: *mut C_tensor,
6232 training_: c_int,
6233 exponential_average_factor_: f64,
6234 epsilon_: f64,
6235 );
6236 pub fn atg_cudnn_convolution(
6237 out__: *mut *mut C_tensor,
6238 self_: *mut C_tensor,
6239 weight_: *mut C_tensor,
6240 padding_data: *const i64,
6241 padding_len: c_int,
6242 stride_data: *const i64,
6243 stride_len: c_int,
6244 dilation_data: *const i64,
6245 dilation_len: c_int,
6246 groups_: i64,
6247 benchmark_: c_int,
6248 deterministic_: c_int,
6249 allow_tf32_: c_int,
6250 );
6251 pub fn atg_cudnn_convolution_add_relu(
6252 out__: *mut *mut C_tensor,
6253 self_: *mut C_tensor,
6254 weight_: *mut C_tensor,
6255 z_: *mut C_tensor,
6256 alpha_: *mut C_scalar,
6257 bias_: *mut C_tensor,
6258 stride_data: *const i64,
6259 stride_len: c_int,
6260 padding_data: *const i64,
6261 padding_len: c_int,
6262 dilation_data: *const i64,
6263 dilation_len: c_int,
6264 groups_: i64,
6265 );
6266 pub fn atg_cudnn_convolution_add_relu_out(
6267 out__: *mut *mut C_tensor,
6268 out_: *mut C_tensor,
6269 self_: *mut C_tensor,
6270 weight_: *mut C_tensor,
6271 z_: *mut C_tensor,
6272 alpha_: *mut C_scalar,
6273 bias_: *mut C_tensor,
6274 stride_data: *const i64,
6275 stride_len: c_int,
6276 padding_data: *const i64,
6277 padding_len: c_int,
6278 dilation_data: *const i64,
6279 dilation_len: c_int,
6280 groups_: i64,
6281 );
6282 pub fn atg_cudnn_convolution_out(
6283 out__: *mut *mut C_tensor,
6284 out_: *mut C_tensor,
6285 self_: *mut C_tensor,
6286 weight_: *mut C_tensor,
6287 padding_data: *const i64,
6288 padding_len: c_int,
6289 stride_data: *const i64,
6290 stride_len: c_int,
6291 dilation_data: *const i64,
6292 dilation_len: c_int,
6293 groups_: i64,
6294 benchmark_: c_int,
6295 deterministic_: c_int,
6296 allow_tf32_: c_int,
6297 );
6298 pub fn atg_cudnn_convolution_relu(
6299 out__: *mut *mut C_tensor,
6300 self_: *mut C_tensor,
6301 weight_: *mut C_tensor,
6302 bias_: *mut C_tensor,
6303 stride_data: *const i64,
6304 stride_len: c_int,
6305 padding_data: *const i64,
6306 padding_len: c_int,
6307 dilation_data: *const i64,
6308 dilation_len: c_int,
6309 groups_: i64,
6310 );
6311 pub fn atg_cudnn_convolution_relu_out(
6312 out__: *mut *mut C_tensor,
6313 out_: *mut C_tensor,
6314 self_: *mut C_tensor,
6315 weight_: *mut C_tensor,
6316 bias_: *mut C_tensor,
6317 stride_data: *const i64,
6318 stride_len: c_int,
6319 padding_data: *const i64,
6320 padding_len: c_int,
6321 dilation_data: *const i64,
6322 dilation_len: c_int,
6323 groups_: i64,
6324 );
6325 pub fn atg_cudnn_convolution_transpose(
6326 out__: *mut *mut C_tensor,
6327 self_: *mut C_tensor,
6328 weight_: *mut C_tensor,
6329 padding_data: *const i64,
6330 padding_len: c_int,
6331 output_padding_data: *const i64,
6332 output_padding_len: c_int,
6333 stride_data: *const i64,
6334 stride_len: c_int,
6335 dilation_data: *const i64,
6336 dilation_len: c_int,
6337 groups_: i64,
6338 benchmark_: c_int,
6339 deterministic_: c_int,
6340 allow_tf32_: c_int,
6341 );
6342 pub fn atg_cudnn_convolution_transpose_out(
6343 out__: *mut *mut C_tensor,
6344 out_: *mut C_tensor,
6345 self_: *mut C_tensor,
6346 weight_: *mut C_tensor,
6347 padding_data: *const i64,
6348 padding_len: c_int,
6349 output_padding_data: *const i64,
6350 output_padding_len: c_int,
6351 stride_data: *const i64,
6352 stride_len: c_int,
6353 dilation_data: *const i64,
6354 dilation_len: c_int,
6355 groups_: i64,
6356 benchmark_: c_int,
6357 deterministic_: c_int,
6358 allow_tf32_: c_int,
6359 );
6360 pub fn atg_cudnn_grid_sampler(
6361 out__: *mut *mut C_tensor,
6362 self_: *mut C_tensor,
6363 grid_: *mut C_tensor,
6364 );
6365 pub fn atg_cudnn_grid_sampler_backward(
6366 out__: *mut *mut C_tensor,
6367 self_: *mut C_tensor,
6368 grid_: *mut C_tensor,
6369 grad_output_: *mut C_tensor,
6370 );
6371 pub fn atg_cudnn_grid_sampler_backward_out(
6372 out__: *mut *mut C_tensor,
6373 out0_: *mut C_tensor,
6374 out1_: *mut C_tensor,
6375 self_: *mut C_tensor,
6376 grid_: *mut C_tensor,
6377 grad_output_: *mut C_tensor,
6378 );
6379 pub fn atg_cudnn_grid_sampler_out(
6380 out__: *mut *mut C_tensor,
6381 out_: *mut C_tensor,
6382 self_: *mut C_tensor,
6383 grid_: *mut C_tensor,
6384 );
6385 pub fn atg_cudnn_is_acceptable(self_: *mut C_tensor) -> c_int;
6386 pub fn atg_cummax(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
6387 pub fn atg_cummax_out(
6388 out__: *mut *mut C_tensor,
6389 values_: *mut C_tensor,
6390 indices_: *mut C_tensor,
6391 self_: *mut C_tensor,
6392 dim_: i64,
6393 );
6394 pub fn atg_cummaxmin_backward(
6395 out__: *mut *mut C_tensor,
6396 grad_: *mut C_tensor,
6397 input_: *mut C_tensor,
6398 indices_: *mut C_tensor,
6399 dim_: i64,
6400 );
6401 pub fn atg_cummin(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
6402 pub fn atg_cummin_out(
6403 out__: *mut *mut C_tensor,
6404 values_: *mut C_tensor,
6405 indices_: *mut C_tensor,
6406 self_: *mut C_tensor,
6407 dim_: i64,
6408 );
6409 pub fn atg_cumprod(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, dtype_: c_int);
6410 pub fn atg_cumprod_(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, dtype_: c_int);
6411 pub fn atg_cumprod_backward(
6412 out__: *mut *mut C_tensor,
6413 grad_: *mut C_tensor,
6414 input_: *mut C_tensor,
6415 dim_: i64,
6416 output_: *mut C_tensor,
6417 );
6418 pub fn atg_cumprod_out(
6419 out__: *mut *mut C_tensor,
6420 out_: *mut C_tensor,
6421 self_: *mut C_tensor,
6422 dim_: i64,
6423 dtype_: c_int,
6424 );
6425 pub fn atg_cumsum(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, dtype_: c_int);
6426 pub fn atg_cumsum_(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, dtype_: c_int);
6427 pub fn atg_cumsum_out(
6428 out__: *mut *mut C_tensor,
6429 out_: *mut C_tensor,
6430 self_: *mut C_tensor,
6431 dim_: i64,
6432 dtype_: c_int,
6433 );
6434 pub fn atg_cumulative_trapezoid(out__: *mut *mut C_tensor, y_: *mut C_tensor, dim_: i64);
6435 pub fn atg_cumulative_trapezoid_x(
6436 out__: *mut *mut C_tensor,
6437 y_: *mut C_tensor,
6438 x_: *mut C_tensor,
6439 dim_: i64,
6440 );
6441 pub fn atg_data(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6442 pub fn atg_deg2rad(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6443 pub fn atg_deg2rad_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6444 pub fn atg_deg2rad_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6445 pub fn atg_dense_dim(self_: *mut C_tensor) -> i64;
6446 pub fn atg_dequantize(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6447 pub fn atg_dequantize_self_out(
6448 out__: *mut *mut C_tensor,
6449 out_: *mut C_tensor,
6450 self_: *mut C_tensor,
6451 );
6452 pub fn atg_dequantize_tensors(
6453 tensors_data: *const *mut C_tensor,
6454 tensors_len: c_int,
6455 ) -> *mut *mut C_tensor;
6456 pub fn atg_dequantize_tensors_out(
6457 out_data: *const *mut C_tensor,
6458 out_len: c_int,
6459 tensors_data: *const *mut C_tensor,
6460 tensors_len: c_int,
6461 );
6462 pub fn atg_det(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6463 pub fn atg_detach(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6464 pub fn atg_detach_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6465 pub fn atg_detach_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6466 pub fn atg_detach_copy_out(
6467 out__: *mut *mut C_tensor,
6468 out_: *mut C_tensor,
6469 self_: *mut C_tensor,
6470 );
6471 pub fn atg_diag(out__: *mut *mut C_tensor, self_: *mut C_tensor, diagonal_: i64);
6472 pub fn atg_diag_embed(
6473 out__: *mut *mut C_tensor,
6474 self_: *mut C_tensor,
6475 offset_: i64,
6476 dim1_: i64,
6477 dim2_: i64,
6478 );
6479 pub fn atg_diag_embed_out(
6480 out__: *mut *mut C_tensor,
6481 out_: *mut C_tensor,
6482 self_: *mut C_tensor,
6483 offset_: i64,
6484 dim1_: i64,
6485 dim2_: i64,
6486 );
6487 pub fn atg_diag_out(
6488 out__: *mut *mut C_tensor,
6489 out_: *mut C_tensor,
6490 self_: *mut C_tensor,
6491 diagonal_: i64,
6492 );
6493 pub fn atg_diagflat(out__: *mut *mut C_tensor, self_: *mut C_tensor, offset_: i64);
6494 pub fn atg_diagonal(
6495 out__: *mut *mut C_tensor,
6496 self_: *mut C_tensor,
6497 offset_: i64,
6498 dim1_: i64,
6499 dim2_: i64,
6500 );
6501 pub fn atg_diagonal_backward(
6502 out__: *mut *mut C_tensor,
6503 grad_output_: *mut C_tensor,
6504 input_sizes_data: *const i64,
6505 input_sizes_len: c_int,
6506 offset_: i64,
6507 dim1_: i64,
6508 dim2_: i64,
6509 );
6510 pub fn atg_diagonal_backward_out(
6511 out__: *mut *mut C_tensor,
6512 out_: *mut C_tensor,
6513 grad_output_: *mut C_tensor,
6514 input_sizes_data: *const i64,
6515 input_sizes_len: c_int,
6516 offset_: i64,
6517 dim1_: i64,
6518 dim2_: i64,
6519 );
6520 pub fn atg_diagonal_copy(
6521 out__: *mut *mut C_tensor,
6522 self_: *mut C_tensor,
6523 offset_: i64,
6524 dim1_: i64,
6525 dim2_: i64,
6526 );
6527 pub fn atg_diagonal_copy_out(
6528 out__: *mut *mut C_tensor,
6529 out_: *mut C_tensor,
6530 self_: *mut C_tensor,
6531 offset_: i64,
6532 dim1_: i64,
6533 dim2_: i64,
6534 );
6535 pub fn atg_diagonal_scatter(
6536 out__: *mut *mut C_tensor,
6537 self_: *mut C_tensor,
6538 src_: *mut C_tensor,
6539 offset_: i64,
6540 dim1_: i64,
6541 dim2_: i64,
6542 );
6543 pub fn atg_diagonal_scatter_out(
6544 out__: *mut *mut C_tensor,
6545 out_: *mut C_tensor,
6546 self_: *mut C_tensor,
6547 src_: *mut C_tensor,
6548 offset_: i64,
6549 dim1_: i64,
6550 dim2_: i64,
6551 );
6552 pub fn atg_diff(
6553 out__: *mut *mut C_tensor,
6554 self_: *mut C_tensor,
6555 n_: i64,
6556 dim_: i64,
6557 prepend_: *mut C_tensor,
6558 append_: *mut C_tensor,
6559 );
6560 pub fn atg_diff_out(
6561 out__: *mut *mut C_tensor,
6562 out_: *mut C_tensor,
6563 self_: *mut C_tensor,
6564 n_: i64,
6565 dim_: i64,
6566 prepend_: *mut C_tensor,
6567 append_: *mut C_tensor,
6568 );
6569 pub fn atg_digamma(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6570 pub fn atg_digamma_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6571 pub fn atg_digamma_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6572 pub fn atg_dist(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6573 pub fn atg_dist_out(
6574 out__: *mut *mut C_tensor,
6575 out_: *mut C_tensor,
6576 self_: *mut C_tensor,
6577 other_: *mut C_tensor,
6578 );
6579 pub fn atg_div(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6580 pub fn atg_div_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6581 pub fn atg_div_out(
6582 out__: *mut *mut C_tensor,
6583 out_: *mut C_tensor,
6584 self_: *mut C_tensor,
6585 other_: *mut C_tensor,
6586 );
6587 pub fn atg_div_out_mode(
6588 out__: *mut *mut C_tensor,
6589 out_: *mut C_tensor,
6590 self_: *mut C_tensor,
6591 other_: *mut C_tensor,
6592 rounding_mode_ptr: *const u8,
6593 rounding_mode_len: c_int,
6594 );
6595 pub fn atg_div_scalar(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
6596 pub fn atg_div_scalar_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
6597 pub fn atg_div_scalar_mode(
6598 out__: *mut *mut C_tensor,
6599 self_: *mut C_tensor,
6600 other_: *mut C_scalar,
6601 rounding_mode_ptr: *const u8,
6602 rounding_mode_len: c_int,
6603 );
6604 pub fn atg_div_scalar_mode_(
6605 out__: *mut *mut C_tensor,
6606 self_: *mut C_tensor,
6607 other_: *mut C_scalar,
6608 rounding_mode_ptr: *const u8,
6609 rounding_mode_len: c_int,
6610 );
6611 pub fn atg_div_scalar_mode_out(
6612 out__: *mut *mut C_tensor,
6613 out_: *mut C_tensor,
6614 self_: *mut C_tensor,
6615 other_: *mut C_scalar,
6616 rounding_mode_ptr: *const u8,
6617 rounding_mode_len: c_int,
6618 );
6619 pub fn atg_div_scalar_out(
6620 out__: *mut *mut C_tensor,
6621 out_: *mut C_tensor,
6622 self_: *mut C_tensor,
6623 other_: *mut C_scalar,
6624 );
6625 pub fn atg_div_tensor_mode(
6626 out__: *mut *mut C_tensor,
6627 self_: *mut C_tensor,
6628 other_: *mut C_tensor,
6629 rounding_mode_ptr: *const u8,
6630 rounding_mode_len: c_int,
6631 );
6632 pub fn atg_div_tensor_mode_(
6633 out__: *mut *mut C_tensor,
6634 self_: *mut C_tensor,
6635 other_: *mut C_tensor,
6636 rounding_mode_ptr: *const u8,
6637 rounding_mode_len: c_int,
6638 );
6639 pub fn atg_divide(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6640 pub fn atg_divide_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6641 pub fn atg_divide_out(
6642 out__: *mut *mut C_tensor,
6643 out_: *mut C_tensor,
6644 self_: *mut C_tensor,
6645 other_: *mut C_tensor,
6646 );
6647 pub fn atg_divide_out_mode(
6648 out__: *mut *mut C_tensor,
6649 out_: *mut C_tensor,
6650 self_: *mut C_tensor,
6651 other_: *mut C_tensor,
6652 rounding_mode_ptr: *const u8,
6653 rounding_mode_len: c_int,
6654 );
6655 pub fn atg_divide_scalar(
6656 out__: *mut *mut C_tensor,
6657 self_: *mut C_tensor,
6658 other_: *mut C_scalar,
6659 );
6660 pub fn atg_divide_scalar_(
6661 out__: *mut *mut C_tensor,
6662 self_: *mut C_tensor,
6663 other_: *mut C_scalar,
6664 );
6665 pub fn atg_divide_scalar_mode(
6666 out__: *mut *mut C_tensor,
6667 self_: *mut C_tensor,
6668 other_: *mut C_scalar,
6669 rounding_mode_ptr: *const u8,
6670 rounding_mode_len: c_int,
6671 );
6672 pub fn atg_divide_scalar_mode_(
6673 out__: *mut *mut C_tensor,
6674 self_: *mut C_tensor,
6675 other_: *mut C_scalar,
6676 rounding_mode_ptr: *const u8,
6677 rounding_mode_len: c_int,
6678 );
6679 pub fn atg_divide_tensor_mode(
6680 out__: *mut *mut C_tensor,
6681 self_: *mut C_tensor,
6682 other_: *mut C_tensor,
6683 rounding_mode_ptr: *const u8,
6684 rounding_mode_len: c_int,
6685 );
6686 pub fn atg_divide_tensor_mode_(
6687 out__: *mut *mut C_tensor,
6688 self_: *mut C_tensor,
6689 other_: *mut C_tensor,
6690 rounding_mode_ptr: *const u8,
6691 rounding_mode_len: c_int,
6692 );
6693 pub fn atg_dot(out__: *mut *mut C_tensor, self_: *mut C_tensor, tensor_: *mut C_tensor);
6694 pub fn atg_dot_out(
6695 out__: *mut *mut C_tensor,
6696 out_: *mut C_tensor,
6697 self_: *mut C_tensor,
6698 tensor_: *mut C_tensor,
6699 );
6700 pub fn atg_dropout(out__: *mut *mut C_tensor, input_: *mut C_tensor, p_: f64, train_: c_int);
6701 pub fn atg_dropout_(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: f64, train_: c_int);
6702 pub fn atg_dsplit(self_: *mut C_tensor, sections_: i64) -> *mut *mut C_tensor;
6703 pub fn atg_dsplit_array(
6704 self_: *mut C_tensor,
6705 indices_data: *const i64,
6706 indices_len: c_int,
6707 ) -> *mut *mut C_tensor;
6708 pub fn atg_dstack(
6709 out__: *mut *mut C_tensor,
6710 tensors_data: *const *mut C_tensor,
6711 tensors_len: c_int,
6712 );
6713 pub fn atg_dstack_out(
6714 out__: *mut *mut C_tensor,
6715 out_: *mut C_tensor,
6716 tensors_data: *const *mut C_tensor,
6717 tensors_len: c_int,
6718 );
6719 pub fn atg_einsum(
6720 out__: *mut *mut C_tensor,
6721 equation_ptr: *const u8,
6722 equation_len: c_int,
6723 tensors_data: *const *mut C_tensor,
6724 tensors_len: c_int,
6725 path_data: *const i64,
6726 path_len: c_int,
6727 );
6728 pub fn atg_elu(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6729 pub fn atg_elu_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6730 pub fn atg_elu_backward(
6731 out__: *mut *mut C_tensor,
6732 grad_output_: *mut C_tensor,
6733 alpha_: *mut C_scalar,
6734 scale_: *mut C_scalar,
6735 input_scale_: *mut C_scalar,
6736 is_result_: c_int,
6737 self_or_result_: *mut C_tensor,
6738 );
6739 pub fn atg_elu_backward_grad_input(
6740 out__: *mut *mut C_tensor,
6741 grad_input_: *mut C_tensor,
6742 grad_output_: *mut C_tensor,
6743 alpha_: *mut C_scalar,
6744 scale_: *mut C_scalar,
6745 input_scale_: *mut C_scalar,
6746 is_result_: c_int,
6747 self_or_result_: *mut C_tensor,
6748 );
6749 pub fn atg_elu_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6750 pub fn atg_embedding(
6751 out__: *mut *mut C_tensor,
6752 weight_: *mut C_tensor,
6753 indices_: *mut C_tensor,
6754 padding_idx_: i64,
6755 scale_grad_by_freq_: c_int,
6756 sparse_: c_int,
6757 );
6758 pub fn atg_embedding_backward(
6759 out__: *mut *mut C_tensor,
6760 grad_: *mut C_tensor,
6761 indices_: *mut C_tensor,
6762 num_weights_: i64,
6763 padding_idx_: i64,
6764 scale_grad_by_freq_: c_int,
6765 sparse_: c_int,
6766 );
6767 pub fn atg_embedding_bag(
6768 out__: *mut *mut C_tensor,
6769 weight_: *mut C_tensor,
6770 indices_: *mut C_tensor,
6771 offsets_: *mut C_tensor,
6772 scale_grad_by_freq_: c_int,
6773 mode_: i64,
6774 sparse_: c_int,
6775 per_sample_weights_: *mut C_tensor,
6776 include_last_offset_: c_int,
6777 );
6778 pub fn atg_embedding_bag_padding_idx(
6779 out__: *mut *mut C_tensor,
6780 weight_: *mut C_tensor,
6781 indices_: *mut C_tensor,
6782 offsets_: *mut C_tensor,
6783 scale_grad_by_freq_: c_int,
6784 mode_: i64,
6785 sparse_: c_int,
6786 per_sample_weights_: *mut C_tensor,
6787 include_last_offset_: c_int,
6788 padding_idx_v: i64,
6789 padding_idx_null: i8,
6790 );
6791 pub fn atg_embedding_dense_backward(
6792 out__: *mut *mut C_tensor,
6793 grad_output_: *mut C_tensor,
6794 indices_: *mut C_tensor,
6795 num_weights_: i64,
6796 padding_idx_: i64,
6797 scale_grad_by_freq_: c_int,
6798 );
6799 pub fn atg_embedding_dense_backward_out(
6800 out__: *mut *mut C_tensor,
6801 out_: *mut C_tensor,
6802 grad_output_: *mut C_tensor,
6803 indices_: *mut C_tensor,
6804 num_weights_: i64,
6805 padding_idx_: i64,
6806 scale_grad_by_freq_: c_int,
6807 );
6808 pub fn atg_embedding_out(
6809 out__: *mut *mut C_tensor,
6810 out_: *mut C_tensor,
6811 weight_: *mut C_tensor,
6812 indices_: *mut C_tensor,
6813 padding_idx_: i64,
6814 scale_grad_by_freq_: c_int,
6815 sparse_: c_int,
6816 );
6817 pub fn atg_embedding_renorm(
6818 out__: *mut *mut C_tensor,
6819 self_: *mut C_tensor,
6820 indices_: *mut C_tensor,
6821 max_norm_: f64,
6822 norm_type_: f64,
6823 );
6824 pub fn atg_embedding_renorm_(
6825 out__: *mut *mut C_tensor,
6826 self_: *mut C_tensor,
6827 indices_: *mut C_tensor,
6828 max_norm_: f64,
6829 norm_type_: f64,
6830 );
6831 pub fn atg_embedding_renorm_out(
6832 out__: *mut *mut C_tensor,
6833 out_: *mut C_tensor,
6834 self_: *mut C_tensor,
6835 indices_: *mut C_tensor,
6836 max_norm_: f64,
6837 norm_type_: f64,
6838 );
6839 pub fn atg_embedding_sparse_backward(
6840 out__: *mut *mut C_tensor,
6841 grad_: *mut C_tensor,
6842 indices_: *mut C_tensor,
6843 num_weights_: i64,
6844 padding_idx_: i64,
6845 scale_grad_by_freq_: c_int,
6846 );
6847 pub fn atg_empty(
6848 out__: *mut *mut C_tensor,
6849 size_data: *const i64,
6850 size_len: c_int,
6851 options_kind: c_int,
6852 options_device: c_int,
6853 );
6854 pub fn atg_empty_like(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6855 pub fn atg_empty_like_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6856 pub fn atg_empty_out(
6857 out__: *mut *mut C_tensor,
6858 out_: *mut C_tensor,
6859 size_data: *const i64,
6860 size_len: c_int,
6861 );
6862 pub fn atg_empty_permuted(
6863 out__: *mut *mut C_tensor,
6864 size_data: *const i64,
6865 size_len: c_int,
6866 physical_layout_data: *const i64,
6867 physical_layout_len: c_int,
6868 options_kind: c_int,
6869 options_device: c_int,
6870 );
6871 pub fn atg_empty_permuted_out(
6872 out__: *mut *mut C_tensor,
6873 out_: *mut C_tensor,
6874 size_data: *const i64,
6875 size_len: c_int,
6876 physical_layout_data: *const i64,
6877 physical_layout_len: c_int,
6878 );
6879 pub fn atg_empty_quantized(
6880 out__: *mut *mut C_tensor,
6881 size_data: *const i64,
6882 size_len: c_int,
6883 qtensor_: *mut C_tensor,
6884 options_kind: c_int,
6885 options_device: c_int,
6886 );
6887 pub fn atg_empty_quantized_out(
6888 out__: *mut *mut C_tensor,
6889 out_: *mut C_tensor,
6890 size_data: *const i64,
6891 size_len: c_int,
6892 qtensor_: *mut C_tensor,
6893 );
6894 pub fn atg_empty_strided(
6895 out__: *mut *mut C_tensor,
6896 size_data: *const i64,
6897 size_len: c_int,
6898 stride_data: *const i64,
6899 stride_len: c_int,
6900 options_kind: c_int,
6901 options_device: c_int,
6902 );
6903 pub fn atg_empty_strided_out(
6904 out__: *mut *mut C_tensor,
6905 out_: *mut C_tensor,
6906 size_data: *const i64,
6907 size_len: c_int,
6908 stride_data: *const i64,
6909 stride_len: c_int,
6910 );
6911 pub fn atg_eq(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
6912 pub fn atg_eq_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
6913 pub fn atg_eq_scalar_out(
6914 out__: *mut *mut C_tensor,
6915 out_: *mut C_tensor,
6916 self_: *mut C_tensor,
6917 other_: *mut C_scalar,
6918 );
6919 pub fn atg_eq_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6920 pub fn atg_eq_tensor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6921 pub fn atg_eq_tensor_out(
6922 out__: *mut *mut C_tensor,
6923 out_: *mut C_tensor,
6924 self_: *mut C_tensor,
6925 other_: *mut C_tensor,
6926 );
6927 pub fn atg_equal(self_: *mut C_tensor, other_: *mut C_tensor) -> c_int;
6928 pub fn atg_erf(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6929 pub fn atg_erf_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6930 pub fn atg_erf_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6931 pub fn atg_erfc(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6932 pub fn atg_erfc_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6933 pub fn atg_erfc_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6934 pub fn atg_erfinv(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6935 pub fn atg_erfinv_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6936 pub fn atg_erfinv_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6937 pub fn atg_exp(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6938 pub fn atg_exp2(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6939 pub fn atg_exp2_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6940 pub fn atg_exp2_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6941 pub fn atg_exp_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6942 pub fn atg_exp_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6943 pub fn atg_expand(
6944 out__: *mut *mut C_tensor,
6945 self_: *mut C_tensor,
6946 size_data: *const i64,
6947 size_len: c_int,
6948 implicit_: c_int,
6949 );
6950 pub fn atg_expand_as(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
6951 pub fn atg_expand_copy(
6952 out__: *mut *mut C_tensor,
6953 self_: *mut C_tensor,
6954 size_data: *const i64,
6955 size_len: c_int,
6956 implicit_: c_int,
6957 );
6958 pub fn atg_expand_copy_out(
6959 out__: *mut *mut C_tensor,
6960 out_: *mut C_tensor,
6961 self_: *mut C_tensor,
6962 size_data: *const i64,
6963 size_len: c_int,
6964 implicit_: c_int,
6965 );
6966 pub fn atg_expm1(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6967 pub fn atg_expm1_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
6968 pub fn atg_expm1_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
6969 pub fn atg_exponential(out__: *mut *mut C_tensor, self_: *mut C_tensor, lambd_: f64);
6970 pub fn atg_exponential_(out__: *mut *mut C_tensor, self_: *mut C_tensor, lambd_: f64);
6971 pub fn atg_exponential_out(
6972 out__: *mut *mut C_tensor,
6973 out_: *mut C_tensor,
6974 self_: *mut C_tensor,
6975 lambd_: f64,
6976 );
6977 pub fn atg_eye(out__: *mut *mut C_tensor, n_: i64, options_kind: c_int, options_device: c_int);
6978 pub fn atg_eye_m(
6979 out__: *mut *mut C_tensor,
6980 n_: i64,
6981 m_: i64,
6982 options_kind: c_int,
6983 options_device: c_int,
6984 );
6985 pub fn atg_eye_m_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, n_: i64, m_: i64);
6986 pub fn atg_eye_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, n_: i64);
6987 pub fn atg_fake_quantize_per_channel_affine(
6988 out__: *mut *mut C_tensor,
6989 self_: *mut C_tensor,
6990 scale_: *mut C_tensor,
6991 zero_point_: *mut C_tensor,
6992 axis_: i64,
6993 quant_min_: i64,
6994 quant_max_: i64,
6995 );
6996 pub fn atg_fake_quantize_per_channel_affine_cachemask(
6997 out__: *mut *mut C_tensor,
6998 self_: *mut C_tensor,
6999 scale_: *mut C_tensor,
7000 zero_point_: *mut C_tensor,
7001 axis_: i64,
7002 quant_min_: i64,
7003 quant_max_: i64,
7004 );
7005 pub fn atg_fake_quantize_per_channel_affine_cachemask_backward(
7006 out__: *mut *mut C_tensor,
7007 grad_: *mut C_tensor,
7008 mask_: *mut C_tensor,
7009 );
7010 pub fn atg_fake_quantize_per_channel_affine_cachemask_out(
7011 out__: *mut *mut C_tensor,
7012 out0_: *mut C_tensor,
7013 out1_: *mut C_tensor,
7014 self_: *mut C_tensor,
7015 scale_: *mut C_tensor,
7016 zero_point_: *mut C_tensor,
7017 axis_: i64,
7018 quant_min_: i64,
7019 quant_max_: i64,
7020 );
7021 pub fn atg_fake_quantize_per_tensor_affine(
7022 out__: *mut *mut C_tensor,
7023 self_: *mut C_tensor,
7024 scale_: f64,
7025 zero_point_: i64,
7026 quant_min_: i64,
7027 quant_max_: i64,
7028 );
7029 pub fn atg_fake_quantize_per_tensor_affine_cachemask(
7030 out__: *mut *mut C_tensor,
7031 self_: *mut C_tensor,
7032 scale_: f64,
7033 zero_point_: i64,
7034 quant_min_: i64,
7035 quant_max_: i64,
7036 );
7037 pub fn atg_fake_quantize_per_tensor_affine_cachemask_backward(
7038 out__: *mut *mut C_tensor,
7039 grad_: *mut C_tensor,
7040 mask_: *mut C_tensor,
7041 );
7042 pub fn atg_fake_quantize_per_tensor_affine_cachemask_out(
7043 out__: *mut *mut C_tensor,
7044 out0_: *mut C_tensor,
7045 out1_: *mut C_tensor,
7046 self_: *mut C_tensor,
7047 scale_: f64,
7048 zero_point_: i64,
7049 quant_min_: i64,
7050 quant_max_: i64,
7051 );
7052 pub fn atg_fake_quantize_per_tensor_affine_tensor_qparams(
7053 out__: *mut *mut C_tensor,
7054 self_: *mut C_tensor,
7055 scale_: *mut C_tensor,
7056 zero_point_: *mut C_tensor,
7057 quant_min_: i64,
7058 quant_max_: i64,
7059 );
7060 pub fn atg_feature_alpha_dropout(
7061 out__: *mut *mut C_tensor,
7062 input_: *mut C_tensor,
7063 p_: f64,
7064 train_: c_int,
7065 );
7066 pub fn atg_feature_alpha_dropout_(
7067 out__: *mut *mut C_tensor,
7068 self_: *mut C_tensor,
7069 p_: f64,
7070 train_: c_int,
7071 );
7072 pub fn atg_feature_dropout(
7073 out__: *mut *mut C_tensor,
7074 input_: *mut C_tensor,
7075 p_: f64,
7076 train_: c_int,
7077 );
7078 pub fn atg_feature_dropout_(
7079 out__: *mut *mut C_tensor,
7080 self_: *mut C_tensor,
7081 p_: f64,
7082 train_: c_int,
7083 );
7084 pub fn atg_fft_fft(
7085 out__: *mut *mut C_tensor,
7086 self_: *mut C_tensor,
7087 n_v: i64,
7088 n_null: i8,
7089 dim_: i64,
7090 norm_ptr: *const u8,
7091 norm_len: c_int,
7092 );
7093 pub fn atg_fft_fft2(
7094 out__: *mut *mut C_tensor,
7095 self_: *mut C_tensor,
7096 s_data: *const i64,
7097 s_len: c_int,
7098 dim_data: *const i64,
7099 dim_len: c_int,
7100 norm_ptr: *const u8,
7101 norm_len: c_int,
7102 );
7103 pub fn atg_fft_fft2_out(
7104 out__: *mut *mut C_tensor,
7105 out_: *mut C_tensor,
7106 self_: *mut C_tensor,
7107 s_data: *const i64,
7108 s_len: c_int,
7109 dim_data: *const i64,
7110 dim_len: c_int,
7111 norm_ptr: *const u8,
7112 norm_len: c_int,
7113 );
7114 pub fn atg_fft_fft_out(
7115 out__: *mut *mut C_tensor,
7116 out_: *mut C_tensor,
7117 self_: *mut C_tensor,
7118 n_v: i64,
7119 n_null: i8,
7120 dim_: i64,
7121 norm_ptr: *const u8,
7122 norm_len: c_int,
7123 );
7124 pub fn atg_fft_fftfreq(
7125 out__: *mut *mut C_tensor,
7126 n_: i64,
7127 d_: f64,
7128 options_kind: c_int,
7129 options_device: c_int,
7130 );
7131 pub fn atg_fft_fftfreq_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, n_: i64, d_: f64);
7132 pub fn atg_fft_fftn(
7133 out__: *mut *mut C_tensor,
7134 self_: *mut C_tensor,
7135 s_data: *const i64,
7136 s_len: c_int,
7137 dim_data: *const i64,
7138 dim_len: c_int,
7139 norm_ptr: *const u8,
7140 norm_len: c_int,
7141 );
7142 pub fn atg_fft_fftn_out(
7143 out__: *mut *mut C_tensor,
7144 out_: *mut C_tensor,
7145 self_: *mut C_tensor,
7146 s_data: *const i64,
7147 s_len: c_int,
7148 dim_data: *const i64,
7149 dim_len: c_int,
7150 norm_ptr: *const u8,
7151 norm_len: c_int,
7152 );
7153 pub fn atg_fft_fftshift(
7154 out__: *mut *mut C_tensor,
7155 self_: *mut C_tensor,
7156 dim_data: *const i64,
7157 dim_len: c_int,
7158 );
7159 pub fn atg_fft_hfft(
7160 out__: *mut *mut C_tensor,
7161 self_: *mut C_tensor,
7162 n_v: i64,
7163 n_null: i8,
7164 dim_: i64,
7165 norm_ptr: *const u8,
7166 norm_len: c_int,
7167 );
7168 pub fn atg_fft_hfft2(
7169 out__: *mut *mut C_tensor,
7170 self_: *mut C_tensor,
7171 s_data: *const i64,
7172 s_len: c_int,
7173 dim_data: *const i64,
7174 dim_len: c_int,
7175 norm_ptr: *const u8,
7176 norm_len: c_int,
7177 );
7178 pub fn atg_fft_hfft2_out(
7179 out__: *mut *mut C_tensor,
7180 out_: *mut C_tensor,
7181 self_: *mut C_tensor,
7182 s_data: *const i64,
7183 s_len: c_int,
7184 dim_data: *const i64,
7185 dim_len: c_int,
7186 norm_ptr: *const u8,
7187 norm_len: c_int,
7188 );
7189 pub fn atg_fft_hfft_out(
7190 out__: *mut *mut C_tensor,
7191 out_: *mut C_tensor,
7192 self_: *mut C_tensor,
7193 n_v: i64,
7194 n_null: i8,
7195 dim_: i64,
7196 norm_ptr: *const u8,
7197 norm_len: c_int,
7198 );
7199 pub fn atg_fft_hfftn(
7200 out__: *mut *mut C_tensor,
7201 self_: *mut C_tensor,
7202 s_data: *const i64,
7203 s_len: c_int,
7204 dim_data: *const i64,
7205 dim_len: c_int,
7206 norm_ptr: *const u8,
7207 norm_len: c_int,
7208 );
7209 pub fn atg_fft_hfftn_out(
7210 out__: *mut *mut C_tensor,
7211 out_: *mut C_tensor,
7212 self_: *mut C_tensor,
7213 s_data: *const i64,
7214 s_len: c_int,
7215 dim_data: *const i64,
7216 dim_len: c_int,
7217 norm_ptr: *const u8,
7218 norm_len: c_int,
7219 );
7220 pub fn atg_fft_ifft(
7221 out__: *mut *mut C_tensor,
7222 self_: *mut C_tensor,
7223 n_v: i64,
7224 n_null: i8,
7225 dim_: i64,
7226 norm_ptr: *const u8,
7227 norm_len: c_int,
7228 );
7229 pub fn atg_fft_ifft2(
7230 out__: *mut *mut C_tensor,
7231 self_: *mut C_tensor,
7232 s_data: *const i64,
7233 s_len: c_int,
7234 dim_data: *const i64,
7235 dim_len: c_int,
7236 norm_ptr: *const u8,
7237 norm_len: c_int,
7238 );
7239 pub fn atg_fft_ifft2_out(
7240 out__: *mut *mut C_tensor,
7241 out_: *mut C_tensor,
7242 self_: *mut C_tensor,
7243 s_data: *const i64,
7244 s_len: c_int,
7245 dim_data: *const i64,
7246 dim_len: c_int,
7247 norm_ptr: *const u8,
7248 norm_len: c_int,
7249 );
7250 pub fn atg_fft_ifft_out(
7251 out__: *mut *mut C_tensor,
7252 out_: *mut C_tensor,
7253 self_: *mut C_tensor,
7254 n_v: i64,
7255 n_null: i8,
7256 dim_: i64,
7257 norm_ptr: *const u8,
7258 norm_len: c_int,
7259 );
7260 pub fn atg_fft_ifftn(
7261 out__: *mut *mut C_tensor,
7262 self_: *mut C_tensor,
7263 s_data: *const i64,
7264 s_len: c_int,
7265 dim_data: *const i64,
7266 dim_len: c_int,
7267 norm_ptr: *const u8,
7268 norm_len: c_int,
7269 );
7270 pub fn atg_fft_ifftn_out(
7271 out__: *mut *mut C_tensor,
7272 out_: *mut C_tensor,
7273 self_: *mut C_tensor,
7274 s_data: *const i64,
7275 s_len: c_int,
7276 dim_data: *const i64,
7277 dim_len: c_int,
7278 norm_ptr: *const u8,
7279 norm_len: c_int,
7280 );
7281 pub fn atg_fft_ifftshift(
7282 out__: *mut *mut C_tensor,
7283 self_: *mut C_tensor,
7284 dim_data: *const i64,
7285 dim_len: c_int,
7286 );
7287 pub fn atg_fft_ihfft(
7288 out__: *mut *mut C_tensor,
7289 self_: *mut C_tensor,
7290 n_v: i64,
7291 n_null: i8,
7292 dim_: i64,
7293 norm_ptr: *const u8,
7294 norm_len: c_int,
7295 );
7296 pub fn atg_fft_ihfft2(
7297 out__: *mut *mut C_tensor,
7298 self_: *mut C_tensor,
7299 s_data: *const i64,
7300 s_len: c_int,
7301 dim_data: *const i64,
7302 dim_len: c_int,
7303 norm_ptr: *const u8,
7304 norm_len: c_int,
7305 );
7306 pub fn atg_fft_ihfft2_out(
7307 out__: *mut *mut C_tensor,
7308 out_: *mut C_tensor,
7309 self_: *mut C_tensor,
7310 s_data: *const i64,
7311 s_len: c_int,
7312 dim_data: *const i64,
7313 dim_len: c_int,
7314 norm_ptr: *const u8,
7315 norm_len: c_int,
7316 );
7317 pub fn atg_fft_ihfft_out(
7318 out__: *mut *mut C_tensor,
7319 out_: *mut C_tensor,
7320 self_: *mut C_tensor,
7321 n_v: i64,
7322 n_null: i8,
7323 dim_: i64,
7324 norm_ptr: *const u8,
7325 norm_len: c_int,
7326 );
7327 pub fn atg_fft_ihfftn(
7328 out__: *mut *mut C_tensor,
7329 self_: *mut C_tensor,
7330 s_data: *const i64,
7331 s_len: c_int,
7332 dim_data: *const i64,
7333 dim_len: c_int,
7334 norm_ptr: *const u8,
7335 norm_len: c_int,
7336 );
7337 pub fn atg_fft_ihfftn_out(
7338 out__: *mut *mut C_tensor,
7339 out_: *mut C_tensor,
7340 self_: *mut C_tensor,
7341 s_data: *const i64,
7342 s_len: c_int,
7343 dim_data: *const i64,
7344 dim_len: c_int,
7345 norm_ptr: *const u8,
7346 norm_len: c_int,
7347 );
7348 pub fn atg_fft_irfft(
7349 out__: *mut *mut C_tensor,
7350 self_: *mut C_tensor,
7351 n_v: i64,
7352 n_null: i8,
7353 dim_: i64,
7354 norm_ptr: *const u8,
7355 norm_len: c_int,
7356 );
7357 pub fn atg_fft_irfft2(
7358 out__: *mut *mut C_tensor,
7359 self_: *mut C_tensor,
7360 s_data: *const i64,
7361 s_len: c_int,
7362 dim_data: *const i64,
7363 dim_len: c_int,
7364 norm_ptr: *const u8,
7365 norm_len: c_int,
7366 );
7367 pub fn atg_fft_irfft2_out(
7368 out__: *mut *mut C_tensor,
7369 out_: *mut C_tensor,
7370 self_: *mut C_tensor,
7371 s_data: *const i64,
7372 s_len: c_int,
7373 dim_data: *const i64,
7374 dim_len: c_int,
7375 norm_ptr: *const u8,
7376 norm_len: c_int,
7377 );
7378 pub fn atg_fft_irfft_out(
7379 out__: *mut *mut C_tensor,
7380 out_: *mut C_tensor,
7381 self_: *mut C_tensor,
7382 n_v: i64,
7383 n_null: i8,
7384 dim_: i64,
7385 norm_ptr: *const u8,
7386 norm_len: c_int,
7387 );
7388 pub fn atg_fft_irfftn(
7389 out__: *mut *mut C_tensor,
7390 self_: *mut C_tensor,
7391 s_data: *const i64,
7392 s_len: c_int,
7393 dim_data: *const i64,
7394 dim_len: c_int,
7395 norm_ptr: *const u8,
7396 norm_len: c_int,
7397 );
7398 pub fn atg_fft_irfftn_out(
7399 out__: *mut *mut C_tensor,
7400 out_: *mut C_tensor,
7401 self_: *mut C_tensor,
7402 s_data: *const i64,
7403 s_len: c_int,
7404 dim_data: *const i64,
7405 dim_len: c_int,
7406 norm_ptr: *const u8,
7407 norm_len: c_int,
7408 );
7409 pub fn atg_fft_rfft(
7410 out__: *mut *mut C_tensor,
7411 self_: *mut C_tensor,
7412 n_v: i64,
7413 n_null: i8,
7414 dim_: i64,
7415 norm_ptr: *const u8,
7416 norm_len: c_int,
7417 );
7418 pub fn atg_fft_rfft2(
7419 out__: *mut *mut C_tensor,
7420 self_: *mut C_tensor,
7421 s_data: *const i64,
7422 s_len: c_int,
7423 dim_data: *const i64,
7424 dim_len: c_int,
7425 norm_ptr: *const u8,
7426 norm_len: c_int,
7427 );
7428 pub fn atg_fft_rfft2_out(
7429 out__: *mut *mut C_tensor,
7430 out_: *mut C_tensor,
7431 self_: *mut C_tensor,
7432 s_data: *const i64,
7433 s_len: c_int,
7434 dim_data: *const i64,
7435 dim_len: c_int,
7436 norm_ptr: *const u8,
7437 norm_len: c_int,
7438 );
7439 pub fn atg_fft_rfft_out(
7440 out__: *mut *mut C_tensor,
7441 out_: *mut C_tensor,
7442 self_: *mut C_tensor,
7443 n_v: i64,
7444 n_null: i8,
7445 dim_: i64,
7446 norm_ptr: *const u8,
7447 norm_len: c_int,
7448 );
7449 pub fn atg_fft_rfftfreq(
7450 out__: *mut *mut C_tensor,
7451 n_: i64,
7452 d_: f64,
7453 options_kind: c_int,
7454 options_device: c_int,
7455 );
7456 pub fn atg_fft_rfftfreq_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, n_: i64, d_: f64);
7457 pub fn atg_fft_rfftn(
7458 out__: *mut *mut C_tensor,
7459 self_: *mut C_tensor,
7460 s_data: *const i64,
7461 s_len: c_int,
7462 dim_data: *const i64,
7463 dim_len: c_int,
7464 norm_ptr: *const u8,
7465 norm_len: c_int,
7466 );
7467 pub fn atg_fft_rfftn_out(
7468 out__: *mut *mut C_tensor,
7469 out_: *mut C_tensor,
7470 self_: *mut C_tensor,
7471 s_data: *const i64,
7472 s_len: c_int,
7473 dim_data: *const i64,
7474 dim_len: c_int,
7475 norm_ptr: *const u8,
7476 norm_len: c_int,
7477 );
7478 pub fn atg_fill(out__: *mut *mut C_tensor, self_: *mut C_tensor, value_: *mut C_scalar);
7479 pub fn atg_fill_(out__: *mut *mut C_tensor, self_: *mut C_tensor, value_: *mut C_scalar);
7480 pub fn atg_fill_diagonal_(
7481 out__: *mut *mut C_tensor,
7482 self_: *mut C_tensor,
7483 fill_value_: *mut C_scalar,
7484 wrap_: c_int,
7485 );
7486 pub fn atg_fill_scalar_out(
7487 out__: *mut *mut C_tensor,
7488 out_: *mut C_tensor,
7489 self_: *mut C_tensor,
7490 value_: *mut C_scalar,
7491 );
7492 pub fn atg_fill_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, value_: *mut C_tensor);
7493 pub fn atg_fill_tensor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, value_: *mut C_tensor);
7494 pub fn atg_fill_tensor_out(
7495 out__: *mut *mut C_tensor,
7496 out_: *mut C_tensor,
7497 self_: *mut C_tensor,
7498 value_: *mut C_tensor,
7499 );
7500 pub fn atg_fix(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7501 pub fn atg_fix_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7502 pub fn atg_fix_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
7503 pub fn atg_flatten(
7504 out__: *mut *mut C_tensor,
7505 self_: *mut C_tensor,
7506 start_dim_: i64,
7507 end_dim_: i64,
7508 );
7509 pub fn atg_flatten_dense_tensors(
7510 out__: *mut *mut C_tensor,
7511 tensors_data: *const *mut C_tensor,
7512 tensors_len: c_int,
7513 );
7514 pub fn atg_flip(
7515 out__: *mut *mut C_tensor,
7516 self_: *mut C_tensor,
7517 dims_data: *const i64,
7518 dims_len: c_int,
7519 );
7520 pub fn atg_flip_out(
7521 out__: *mut *mut C_tensor,
7522 out_: *mut C_tensor,
7523 self_: *mut C_tensor,
7524 dims_data: *const i64,
7525 dims_len: c_int,
7526 );
7527 pub fn atg_fliplr(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7528 pub fn atg_flipud(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7529 pub fn atg_float_power(
7530 out__: *mut *mut C_tensor,
7531 self_: *mut C_tensor,
7532 exponent_: *mut C_tensor,
7533 );
7534 pub fn atg_float_power_(
7535 out__: *mut *mut C_tensor,
7536 self_: *mut C_tensor,
7537 exponent_: *mut C_scalar,
7538 );
7539 pub fn atg_float_power_scalar(
7540 out__: *mut *mut C_tensor,
7541 self_scalar_: *mut C_scalar,
7542 exponent_: *mut C_tensor,
7543 );
7544 pub fn atg_float_power_scalar_out(
7545 out__: *mut *mut C_tensor,
7546 out_: *mut C_tensor,
7547 self_scalar_: *mut C_scalar,
7548 exponent_: *mut C_tensor,
7549 );
7550 pub fn atg_float_power_tensor_(
7551 out__: *mut *mut C_tensor,
7552 self_: *mut C_tensor,
7553 exponent_: *mut C_tensor,
7554 );
7555 pub fn atg_float_power_tensor_scalar(
7556 out__: *mut *mut C_tensor,
7557 self_: *mut C_tensor,
7558 exponent_: *mut C_scalar,
7559 );
7560 pub fn atg_float_power_tensor_scalar_out(
7561 out__: *mut *mut C_tensor,
7562 out_: *mut C_tensor,
7563 self_: *mut C_tensor,
7564 exponent_: *mut C_scalar,
7565 );
7566 pub fn atg_float_power_tensor_tensor_out(
7567 out__: *mut *mut C_tensor,
7568 out_: *mut C_tensor,
7569 self_: *mut C_tensor,
7570 exponent_: *mut C_tensor,
7571 );
7572 pub fn atg_floor(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7573 pub fn atg_floor_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7574 pub fn atg_floor_divide(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
7575 pub fn atg_floor_divide_(
7576 out__: *mut *mut C_tensor,
7577 self_: *mut C_tensor,
7578 other_: *mut C_tensor,
7579 );
7580 pub fn atg_floor_divide_out(
7581 out__: *mut *mut C_tensor,
7582 out_: *mut C_tensor,
7583 self_: *mut C_tensor,
7584 other_: *mut C_tensor,
7585 );
7586 pub fn atg_floor_divide_scalar(
7587 out__: *mut *mut C_tensor,
7588 self_: *mut C_tensor,
7589 other_: *mut C_scalar,
7590 );
7591 pub fn atg_floor_divide_scalar_(
7592 out__: *mut *mut C_tensor,
7593 self_: *mut C_tensor,
7594 other_: *mut C_scalar,
7595 );
7596 pub fn atg_floor_divide_scalar_out(
7597 out__: *mut *mut C_tensor,
7598 out_: *mut C_tensor,
7599 self_: *mut C_tensor,
7600 other_: *mut C_scalar,
7601 );
7602 pub fn atg_floor_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
7603 pub fn atg_fmax(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
7604 pub fn atg_fmax_out(
7605 out__: *mut *mut C_tensor,
7606 out_: *mut C_tensor,
7607 self_: *mut C_tensor,
7608 other_: *mut C_tensor,
7609 );
7610 pub fn atg_fmin(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
7611 pub fn atg_fmin_out(
7612 out__: *mut *mut C_tensor,
7613 out_: *mut C_tensor,
7614 self_: *mut C_tensor,
7615 other_: *mut C_tensor,
7616 );
7617 pub fn atg_fmod(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
7618 pub fn atg_fmod_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
7619 pub fn atg_fmod_scalar_out(
7620 out__: *mut *mut C_tensor,
7621 out_: *mut C_tensor,
7622 self_: *mut C_tensor,
7623 other_: *mut C_scalar,
7624 );
7625 pub fn atg_fmod_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
7626 pub fn atg_fmod_tensor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
7627 pub fn atg_fmod_tensor_out(
7628 out__: *mut *mut C_tensor,
7629 out_: *mut C_tensor,
7630 self_: *mut C_tensor,
7631 other_: *mut C_tensor,
7632 );
7633 pub fn atg_frac(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7634 pub fn atg_frac_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7635 pub fn atg_frac_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
7636 pub fn atg_fractional_max_pool2d(
7637 out__: *mut *mut C_tensor,
7638 self_: *mut C_tensor,
7639 kernel_size_data: *const i64,
7640 kernel_size_len: c_int,
7641 output_size_data: *const i64,
7642 output_size_len: c_int,
7643 random_samples_: *mut C_tensor,
7644 );
7645 pub fn atg_fractional_max_pool2d_backward(
7646 out__: *mut *mut C_tensor,
7647 grad_output_: *mut C_tensor,
7648 self_: *mut C_tensor,
7649 kernel_size_data: *const i64,
7650 kernel_size_len: c_int,
7651 output_size_data: *const i64,
7652 output_size_len: c_int,
7653 indices_: *mut C_tensor,
7654 );
7655 pub fn atg_fractional_max_pool2d_backward_grad_input(
7656 out__: *mut *mut C_tensor,
7657 grad_input_: *mut C_tensor,
7658 grad_output_: *mut C_tensor,
7659 self_: *mut C_tensor,
7660 kernel_size_data: *const i64,
7661 kernel_size_len: c_int,
7662 output_size_data: *const i64,
7663 output_size_len: c_int,
7664 indices_: *mut C_tensor,
7665 );
7666 pub fn atg_fractional_max_pool2d_output(
7667 out__: *mut *mut C_tensor,
7668 output_: *mut C_tensor,
7669 indices_: *mut C_tensor,
7670 self_: *mut C_tensor,
7671 kernel_size_data: *const i64,
7672 kernel_size_len: c_int,
7673 output_size_data: *const i64,
7674 output_size_len: c_int,
7675 random_samples_: *mut C_tensor,
7676 );
7677 pub fn atg_fractional_max_pool3d(
7678 out__: *mut *mut C_tensor,
7679 self_: *mut C_tensor,
7680 kernel_size_data: *const i64,
7681 kernel_size_len: c_int,
7682 output_size_data: *const i64,
7683 output_size_len: c_int,
7684 random_samples_: *mut C_tensor,
7685 );
7686 pub fn atg_fractional_max_pool3d_backward(
7687 out__: *mut *mut C_tensor,
7688 grad_output_: *mut C_tensor,
7689 self_: *mut C_tensor,
7690 kernel_size_data: *const i64,
7691 kernel_size_len: c_int,
7692 output_size_data: *const i64,
7693 output_size_len: c_int,
7694 indices_: *mut C_tensor,
7695 );
7696 pub fn atg_fractional_max_pool3d_backward_grad_input(
7697 out__: *mut *mut C_tensor,
7698 grad_input_: *mut C_tensor,
7699 grad_output_: *mut C_tensor,
7700 self_: *mut C_tensor,
7701 kernel_size_data: *const i64,
7702 kernel_size_len: c_int,
7703 output_size_data: *const i64,
7704 output_size_len: c_int,
7705 indices_: *mut C_tensor,
7706 );
7707 pub fn atg_fractional_max_pool3d_output(
7708 out__: *mut *mut C_tensor,
7709 output_: *mut C_tensor,
7710 indices_: *mut C_tensor,
7711 self_: *mut C_tensor,
7712 kernel_size_data: *const i64,
7713 kernel_size_len: c_int,
7714 output_size_data: *const i64,
7715 output_size_len: c_int,
7716 random_samples_: *mut C_tensor,
7717 );
7718 pub fn atg_frexp(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7719 pub fn atg_frexp_tensor_out(
7720 out__: *mut *mut C_tensor,
7721 mantissa_: *mut C_tensor,
7722 exponent_: *mut C_tensor,
7723 self_: *mut C_tensor,
7724 );
7725 pub fn atg_frobenius_norm(
7726 out__: *mut *mut C_tensor,
7727 self_: *mut C_tensor,
7728 dim_data: *const i64,
7729 dim_len: c_int,
7730 keepdim_: c_int,
7731 );
7732 pub fn atg_frobenius_norm_out(
7733 out__: *mut *mut C_tensor,
7734 out_: *mut C_tensor,
7735 self_: *mut C_tensor,
7736 dim_data: *const i64,
7737 dim_len: c_int,
7738 keepdim_: c_int,
7739 );
7740 pub fn atg_from_file(
7741 out__: *mut *mut C_tensor,
7742 filename_ptr: *const u8,
7743 filename_len: c_int,
7744 shared_: c_int,
7745 size_v: i64,
7746 size_null: i8,
7747 options_kind: c_int,
7748 options_device: c_int,
7749 );
7750 pub fn atg_from_file_out(
7751 out__: *mut *mut C_tensor,
7752 out_: *mut C_tensor,
7753 filename_ptr: *const u8,
7754 filename_len: c_int,
7755 shared_: c_int,
7756 size_v: i64,
7757 size_null: i8,
7758 );
7759 pub fn atg_full(
7760 out__: *mut *mut C_tensor,
7761 size_data: *const i64,
7762 size_len: c_int,
7763 fill_value_: *mut C_scalar,
7764 options_kind: c_int,
7765 options_device: c_int,
7766 );
7767 pub fn atg_full_like(
7768 out__: *mut *mut C_tensor,
7769 self_: *mut C_tensor,
7770 fill_value_: *mut C_scalar,
7771 );
7772 pub fn atg_full_like_out(
7773 out__: *mut *mut C_tensor,
7774 out_: *mut C_tensor,
7775 self_: *mut C_tensor,
7776 fill_value_: *mut C_scalar,
7777 );
7778 pub fn atg_full_out(
7779 out__: *mut *mut C_tensor,
7780 out_: *mut C_tensor,
7781 size_data: *const i64,
7782 size_len: c_int,
7783 fill_value_: *mut C_scalar,
7784 );
7785 pub fn atg_fused_moving_avg_obs_fake_quant(
7786 out__: *mut *mut C_tensor,
7787 self_: *mut C_tensor,
7788 observer_on_: *mut C_tensor,
7789 fake_quant_on_: *mut C_tensor,
7790 running_min_: *mut C_tensor,
7791 running_max_: *mut C_tensor,
7792 scale_: *mut C_tensor,
7793 zero_point_: *mut C_tensor,
7794 averaging_const_: f64,
7795 quant_min_: i64,
7796 quant_max_: i64,
7797 ch_axis_: i64,
7798 per_row_fake_quant_: c_int,
7799 symmetric_quant_: c_int,
7800 );
7801 pub fn atg_gather(
7802 out__: *mut *mut C_tensor,
7803 self_: *mut C_tensor,
7804 dim_: i64,
7805 index_: *mut C_tensor,
7806 sparse_grad_: c_int,
7807 );
7808 pub fn atg_gather_backward(
7809 out__: *mut *mut C_tensor,
7810 grad_: *mut C_tensor,
7811 self_: *mut C_tensor,
7812 dim_: i64,
7813 index_: *mut C_tensor,
7814 sparse_grad_: c_int,
7815 );
7816 pub fn atg_gather_out(
7817 out__: *mut *mut C_tensor,
7818 out_: *mut C_tensor,
7819 self_: *mut C_tensor,
7820 dim_: i64,
7821 index_: *mut C_tensor,
7822 sparse_grad_: c_int,
7823 );
7824 pub fn atg_gcd(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
7825 pub fn atg_gcd_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
7826 pub fn atg_gcd_out(
7827 out__: *mut *mut C_tensor,
7828 out_: *mut C_tensor,
7829 self_: *mut C_tensor,
7830 other_: *mut C_tensor,
7831 );
7832 pub fn atg_ge(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
7833 pub fn atg_ge_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
7834 pub fn atg_ge_scalar_out(
7835 out__: *mut *mut C_tensor,
7836 out_: *mut C_tensor,
7837 self_: *mut C_tensor,
7838 other_: *mut C_scalar,
7839 );
7840 pub fn atg_ge_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
7841 pub fn atg_ge_tensor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
7842 pub fn atg_ge_tensor_out(
7843 out__: *mut *mut C_tensor,
7844 out_: *mut C_tensor,
7845 self_: *mut C_tensor,
7846 other_: *mut C_tensor,
7847 );
7848 pub fn atg_gelu(
7849 out__: *mut *mut C_tensor,
7850 self_: *mut C_tensor,
7851 approximate_ptr: *const u8,
7852 approximate_len: c_int,
7853 );
7854 pub fn atg_gelu_(
7855 out__: *mut *mut C_tensor,
7856 self_: *mut C_tensor,
7857 approximate_ptr: *const u8,
7858 approximate_len: c_int,
7859 );
7860 pub fn atg_gelu_backward(
7861 out__: *mut *mut C_tensor,
7862 grad_output_: *mut C_tensor,
7863 self_: *mut C_tensor,
7864 approximate_ptr: *const u8,
7865 approximate_len: c_int,
7866 );
7867 pub fn atg_gelu_backward_grad_input(
7868 out__: *mut *mut C_tensor,
7869 grad_input_: *mut C_tensor,
7870 grad_output_: *mut C_tensor,
7871 self_: *mut C_tensor,
7872 approximate_ptr: *const u8,
7873 approximate_len: c_int,
7874 );
7875 pub fn atg_gelu_out(
7876 out__: *mut *mut C_tensor,
7877 out_: *mut C_tensor,
7878 self_: *mut C_tensor,
7879 approximate_ptr: *const u8,
7880 approximate_len: c_int,
7881 );
7882 pub fn atg_geometric(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: f64);
7883 pub fn atg_geometric_(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: f64);
7884 pub fn atg_geometric_out(
7885 out__: *mut *mut C_tensor,
7886 out_: *mut C_tensor,
7887 self_: *mut C_tensor,
7888 p_: f64,
7889 );
7890 pub fn atg_geqrf(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7891 pub fn atg_geqrf_a(
7892 out__: *mut *mut C_tensor,
7893 a_: *mut C_tensor,
7894 tau_: *mut C_tensor,
7895 self_: *mut C_tensor,
7896 );
7897 pub fn atg_ger(out__: *mut *mut C_tensor, self_: *mut C_tensor, vec2_: *mut C_tensor);
7898 pub fn atg_ger_out(
7899 out__: *mut *mut C_tensor,
7900 out_: *mut C_tensor,
7901 self_: *mut C_tensor,
7902 vec2_: *mut C_tensor,
7903 );
7904 pub fn atg_get_device(self_: *mut C_tensor) -> i64;
7905 pub fn atg_glu(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
7906 pub fn atg_glu_backward(
7907 out__: *mut *mut C_tensor,
7908 grad_output_: *mut C_tensor,
7909 self_: *mut C_tensor,
7910 dim_: i64,
7911 );
7912 pub fn atg_glu_backward_grad_input(
7913 out__: *mut *mut C_tensor,
7914 grad_input_: *mut C_tensor,
7915 grad_output_: *mut C_tensor,
7916 self_: *mut C_tensor,
7917 dim_: i64,
7918 );
7919 pub fn atg_glu_backward_jvp(
7920 out__: *mut *mut C_tensor,
7921 grad_x_: *mut C_tensor,
7922 grad_glu_: *mut C_tensor,
7923 x_: *mut C_tensor,
7924 dgrad_glu_: *mut C_tensor,
7925 dx_: *mut C_tensor,
7926 dim_: i64,
7927 );
7928 pub fn atg_glu_backward_jvp_out(
7929 out__: *mut *mut C_tensor,
7930 out_: *mut C_tensor,
7931 grad_x_: *mut C_tensor,
7932 grad_glu_: *mut C_tensor,
7933 x_: *mut C_tensor,
7934 dgrad_glu_: *mut C_tensor,
7935 dx_: *mut C_tensor,
7936 dim_: i64,
7937 );
7938 pub fn atg_glu_jvp(
7939 out__: *mut *mut C_tensor,
7940 glu_: *mut C_tensor,
7941 x_: *mut C_tensor,
7942 dx_: *mut C_tensor,
7943 dim_: i64,
7944 );
7945 pub fn atg_glu_jvp_out(
7946 out__: *mut *mut C_tensor,
7947 out_: *mut C_tensor,
7948 glu_: *mut C_tensor,
7949 x_: *mut C_tensor,
7950 dx_: *mut C_tensor,
7951 dim_: i64,
7952 );
7953 pub fn atg_glu_out(
7954 out__: *mut *mut C_tensor,
7955 out_: *mut C_tensor,
7956 self_: *mut C_tensor,
7957 dim_: i64,
7958 );
7959 pub fn atg_grad(out__: *mut *mut C_tensor, self_: *mut C_tensor);
7960 pub fn atg_greater(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
7961 pub fn atg_greater_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
7962 pub fn atg_greater_equal(
7963 out__: *mut *mut C_tensor,
7964 self_: *mut C_tensor,
7965 other_: *mut C_scalar,
7966 );
7967 pub fn atg_greater_equal_(
7968 out__: *mut *mut C_tensor,
7969 self_: *mut C_tensor,
7970 other_: *mut C_scalar,
7971 );
7972 pub fn atg_greater_equal_scalar_out(
7973 out__: *mut *mut C_tensor,
7974 out_: *mut C_tensor,
7975 self_: *mut C_tensor,
7976 other_: *mut C_scalar,
7977 );
7978 pub fn atg_greater_equal_tensor(
7979 out__: *mut *mut C_tensor,
7980 self_: *mut C_tensor,
7981 other_: *mut C_tensor,
7982 );
7983 pub fn atg_greater_equal_tensor_(
7984 out__: *mut *mut C_tensor,
7985 self_: *mut C_tensor,
7986 other_: *mut C_tensor,
7987 );
7988 pub fn atg_greater_equal_tensor_out(
7989 out__: *mut *mut C_tensor,
7990 out_: *mut C_tensor,
7991 self_: *mut C_tensor,
7992 other_: *mut C_tensor,
7993 );
7994 pub fn atg_greater_scalar_out(
7995 out__: *mut *mut C_tensor,
7996 out_: *mut C_tensor,
7997 self_: *mut C_tensor,
7998 other_: *mut C_scalar,
7999 );
8000 pub fn atg_greater_tensor(
8001 out__: *mut *mut C_tensor,
8002 self_: *mut C_tensor,
8003 other_: *mut C_tensor,
8004 );
8005 pub fn atg_greater_tensor_(
8006 out__: *mut *mut C_tensor,
8007 self_: *mut C_tensor,
8008 other_: *mut C_tensor,
8009 );
8010 pub fn atg_greater_tensor_out(
8011 out__: *mut *mut C_tensor,
8012 out_: *mut C_tensor,
8013 self_: *mut C_tensor,
8014 other_: *mut C_tensor,
8015 );
8016 pub fn atg_grid_sampler(
8017 out__: *mut *mut C_tensor,
8018 input_: *mut C_tensor,
8019 grid_: *mut C_tensor,
8020 interpolation_mode_: i64,
8021 padding_mode_: i64,
8022 align_corners_: c_int,
8023 );
8024 pub fn atg_grid_sampler_2d(
8025 out__: *mut *mut C_tensor,
8026 input_: *mut C_tensor,
8027 grid_: *mut C_tensor,
8028 interpolation_mode_: i64,
8029 padding_mode_: i64,
8030 align_corners_: c_int,
8031 );
8032 pub fn atg_grid_sampler_2d_out(
8033 out__: *mut *mut C_tensor,
8034 out_: *mut C_tensor,
8035 input_: *mut C_tensor,
8036 grid_: *mut C_tensor,
8037 interpolation_mode_: i64,
8038 padding_mode_: i64,
8039 align_corners_: c_int,
8040 );
8041 pub fn atg_grid_sampler_3d(
8042 out__: *mut *mut C_tensor,
8043 input_: *mut C_tensor,
8044 grid_: *mut C_tensor,
8045 interpolation_mode_: i64,
8046 padding_mode_: i64,
8047 align_corners_: c_int,
8048 );
8049 pub fn atg_grid_sampler_3d_out(
8050 out__: *mut *mut C_tensor,
8051 out_: *mut C_tensor,
8052 input_: *mut C_tensor,
8053 grid_: *mut C_tensor,
8054 interpolation_mode_: i64,
8055 padding_mode_: i64,
8056 align_corners_: c_int,
8057 );
8058 pub fn atg_group_norm(
8059 out__: *mut *mut C_tensor,
8060 input_: *mut C_tensor,
8061 num_groups_: i64,
8062 weight_: *mut C_tensor,
8063 bias_: *mut C_tensor,
8064 eps_: f64,
8065 cudnn_enabled_: c_int,
8066 );
8067 pub fn atg_gru(
8068 out__: *mut *mut C_tensor,
8069 input_: *mut C_tensor,
8070 hx_: *mut C_tensor,
8071 params_data: *const *mut C_tensor,
8072 params_len: c_int,
8073 has_biases_: c_int,
8074 num_layers_: i64,
8075 dropout_: f64,
8076 train_: c_int,
8077 bidirectional_: c_int,
8078 batch_first_: c_int,
8079 );
8080 pub fn atg_gru_cell(
8081 out__: *mut *mut C_tensor,
8082 input_: *mut C_tensor,
8083 hx_: *mut C_tensor,
8084 w_ih_: *mut C_tensor,
8085 w_hh_: *mut C_tensor,
8086 b_ih_: *mut C_tensor,
8087 b_hh_: *mut C_tensor,
8088 );
8089 pub fn atg_gru_data(
8090 out__: *mut *mut C_tensor,
8091 data_: *mut C_tensor,
8092 batch_sizes_: *mut C_tensor,
8093 hx_: *mut C_tensor,
8094 params_data: *const *mut C_tensor,
8095 params_len: c_int,
8096 has_biases_: c_int,
8097 num_layers_: i64,
8098 dropout_: f64,
8099 train_: c_int,
8100 bidirectional_: c_int,
8101 );
8102 pub fn atg_gt(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
8103 pub fn atg_gt_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
8104 pub fn atg_gt_scalar_out(
8105 out__: *mut *mut C_tensor,
8106 out_: *mut C_tensor,
8107 self_: *mut C_tensor,
8108 other_: *mut C_scalar,
8109 );
8110 pub fn atg_gt_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8111 pub fn atg_gt_tensor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8112 pub fn atg_gt_tensor_out(
8113 out__: *mut *mut C_tensor,
8114 out_: *mut C_tensor,
8115 self_: *mut C_tensor,
8116 other_: *mut C_tensor,
8117 );
8118 pub fn atg_hamming_window(
8119 out__: *mut *mut C_tensor,
8120 window_length_: i64,
8121 options_kind: c_int,
8122 options_device: c_int,
8123 );
8124 pub fn atg_hamming_window_out(
8125 out__: *mut *mut C_tensor,
8126 out_: *mut C_tensor,
8127 window_length_: i64,
8128 );
8129 pub fn atg_hamming_window_periodic(
8130 out__: *mut *mut C_tensor,
8131 window_length_: i64,
8132 periodic_: c_int,
8133 options_kind: c_int,
8134 options_device: c_int,
8135 );
8136 pub fn atg_hamming_window_periodic_alpha(
8137 out__: *mut *mut C_tensor,
8138 window_length_: i64,
8139 periodic_: c_int,
8140 alpha_: f64,
8141 options_kind: c_int,
8142 options_device: c_int,
8143 );
8144 pub fn atg_hamming_window_periodic_alpha_beta(
8145 out__: *mut *mut C_tensor,
8146 window_length_: i64,
8147 periodic_: c_int,
8148 alpha_: f64,
8149 beta_: f64,
8150 options_kind: c_int,
8151 options_device: c_int,
8152 );
8153 pub fn atg_hamming_window_periodic_alpha_beta_out(
8154 out__: *mut *mut C_tensor,
8155 out_: *mut C_tensor,
8156 window_length_: i64,
8157 periodic_: c_int,
8158 alpha_: f64,
8159 beta_: f64,
8160 );
8161 pub fn atg_hamming_window_periodic_alpha_out(
8162 out__: *mut *mut C_tensor,
8163 out_: *mut C_tensor,
8164 window_length_: i64,
8165 periodic_: c_int,
8166 alpha_: f64,
8167 );
8168 pub fn atg_hamming_window_periodic_out(
8169 out__: *mut *mut C_tensor,
8170 out_: *mut C_tensor,
8171 window_length_: i64,
8172 periodic_: c_int,
8173 );
8174 pub fn atg_hann_window(
8175 out__: *mut *mut C_tensor,
8176 window_length_: i64,
8177 options_kind: c_int,
8178 options_device: c_int,
8179 );
8180 pub fn atg_hann_window_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, window_length_: i64);
8181 pub fn atg_hann_window_periodic(
8182 out__: *mut *mut C_tensor,
8183 window_length_: i64,
8184 periodic_: c_int,
8185 options_kind: c_int,
8186 options_device: c_int,
8187 );
8188 pub fn atg_hann_window_periodic_out(
8189 out__: *mut *mut C_tensor,
8190 out_: *mut C_tensor,
8191 window_length_: i64,
8192 periodic_: c_int,
8193 );
8194 pub fn atg_hardshrink(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8195 pub fn atg_hardshrink_backward(
8196 out__: *mut *mut C_tensor,
8197 grad_out_: *mut C_tensor,
8198 self_: *mut C_tensor,
8199 lambd_: *mut C_scalar,
8200 );
8201 pub fn atg_hardshrink_backward_grad_input(
8202 out__: *mut *mut C_tensor,
8203 grad_input_: *mut C_tensor,
8204 grad_out_: *mut C_tensor,
8205 self_: *mut C_tensor,
8206 lambd_: *mut C_scalar,
8207 );
8208 pub fn atg_hardshrink_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8209 pub fn atg_hardsigmoid(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8210 pub fn atg_hardsigmoid_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8211 pub fn atg_hardsigmoid_backward(
8212 out__: *mut *mut C_tensor,
8213 grad_output_: *mut C_tensor,
8214 self_: *mut C_tensor,
8215 );
8216 pub fn atg_hardsigmoid_backward_grad_input(
8217 out__: *mut *mut C_tensor,
8218 grad_input_: *mut C_tensor,
8219 grad_output_: *mut C_tensor,
8220 self_: *mut C_tensor,
8221 );
8222 pub fn atg_hardsigmoid_out(
8223 out__: *mut *mut C_tensor,
8224 out_: *mut C_tensor,
8225 self_: *mut C_tensor,
8226 );
8227 pub fn atg_hardswish(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8228 pub fn atg_hardswish_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8229 pub fn atg_hardswish_backward(
8230 out__: *mut *mut C_tensor,
8231 grad_output_: *mut C_tensor,
8232 self_: *mut C_tensor,
8233 );
8234 pub fn atg_hardswish_backward_out(
8235 out__: *mut *mut C_tensor,
8236 out_: *mut C_tensor,
8237 grad_output_: *mut C_tensor,
8238 self_: *mut C_tensor,
8239 );
8240 pub fn atg_hardswish_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8241 pub fn atg_hardtanh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8242 pub fn atg_hardtanh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8243 pub fn atg_hardtanh_backward(
8244 out__: *mut *mut C_tensor,
8245 grad_output_: *mut C_tensor,
8246 self_: *mut C_tensor,
8247 min_val_: *mut C_scalar,
8248 max_val_: *mut C_scalar,
8249 );
8250 pub fn atg_hardtanh_backward_grad_input(
8251 out__: *mut *mut C_tensor,
8252 grad_input_: *mut C_tensor,
8253 grad_output_: *mut C_tensor,
8254 self_: *mut C_tensor,
8255 min_val_: *mut C_scalar,
8256 max_val_: *mut C_scalar,
8257 );
8258 pub fn atg_hardtanh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8259 pub fn atg_hash_tensor(
8260 out__: *mut *mut C_tensor,
8261 self_: *mut C_tensor,
8262 dim_data: *const i64,
8263 dim_len: c_int,
8264 keepdim_: c_int,
8265 mode_: i64,
8266 );
8267 pub fn atg_hash_tensor_out(
8268 out__: *mut *mut C_tensor,
8269 out_: *mut C_tensor,
8270 self_: *mut C_tensor,
8271 dim_data: *const i64,
8272 dim_len: c_int,
8273 keepdim_: c_int,
8274 mode_: i64,
8275 );
8276 pub fn atg_heaviside(out__: *mut *mut C_tensor, self_: *mut C_tensor, values_: *mut C_tensor);
8277 pub fn atg_heaviside_(out__: *mut *mut C_tensor, self_: *mut C_tensor, values_: *mut C_tensor);
8278 pub fn atg_heaviside_out(
8279 out__: *mut *mut C_tensor,
8280 out_: *mut C_tensor,
8281 self_: *mut C_tensor,
8282 values_: *mut C_tensor,
8283 );
8284 pub fn atg_hinge_embedding_loss(
8285 out__: *mut *mut C_tensor,
8286 self_: *mut C_tensor,
8287 target_: *mut C_tensor,
8288 margin_: f64,
8289 reduction_: i64,
8290 );
8291 pub fn atg_histc(out__: *mut *mut C_tensor, self_: *mut C_tensor, bins_: i64);
8292 pub fn atg_histc_out(
8293 out__: *mut *mut C_tensor,
8294 out_: *mut C_tensor,
8295 self_: *mut C_tensor,
8296 bins_: i64,
8297 );
8298 pub fn atg_histogram(
8299 out__: *mut *mut C_tensor,
8300 self_: *mut C_tensor,
8301 bins_: *mut C_tensor,
8302 weight_: *mut C_tensor,
8303 density_: c_int,
8304 );
8305 pub fn atg_histogram_bin_ct(
8306 out__: *mut *mut C_tensor,
8307 self_: *mut C_tensor,
8308 bins_: i64,
8309 range_data: *const f64,
8310 range_len: c_int,
8311 weight_: *mut C_tensor,
8312 density_: c_int,
8313 );
8314 pub fn atg_histogram_bin_ct_out(
8315 out__: *mut *mut C_tensor,
8316 hist_: *mut C_tensor,
8317 bin_edges_: *mut C_tensor,
8318 self_: *mut C_tensor,
8319 bins_: i64,
8320 range_data: *const f64,
8321 range_len: c_int,
8322 weight_: *mut C_tensor,
8323 density_: c_int,
8324 );
8325 pub fn atg_histogram_bins_tensor_out(
8326 out__: *mut *mut C_tensor,
8327 hist_: *mut C_tensor,
8328 bin_edges_: *mut C_tensor,
8329 self_: *mut C_tensor,
8330 bins_: *mut C_tensor,
8331 weight_: *mut C_tensor,
8332 density_: c_int,
8333 );
8334 pub fn atg_hsplit(self_: *mut C_tensor, sections_: i64) -> *mut *mut C_tensor;
8335 pub fn atg_hsplit_array(
8336 self_: *mut C_tensor,
8337 indices_data: *const i64,
8338 indices_len: c_int,
8339 ) -> *mut *mut C_tensor;
8340 pub fn atg_hspmm(out__: *mut *mut C_tensor, mat1_: *mut C_tensor, mat2_: *mut C_tensor);
8341 pub fn atg_hspmm_out(
8342 out__: *mut *mut C_tensor,
8343 out_: *mut C_tensor,
8344 mat1_: *mut C_tensor,
8345 mat2_: *mut C_tensor,
8346 );
8347 pub fn atg_hstack(
8348 out__: *mut *mut C_tensor,
8349 tensors_data: *const *mut C_tensor,
8350 tensors_len: c_int,
8351 );
8352 pub fn atg_hstack_out(
8353 out__: *mut *mut C_tensor,
8354 out_: *mut C_tensor,
8355 tensors_data: *const *mut C_tensor,
8356 tensors_len: c_int,
8357 );
8358 pub fn atg_huber_loss(
8359 out__: *mut *mut C_tensor,
8360 self_: *mut C_tensor,
8361 target_: *mut C_tensor,
8362 reduction_: i64,
8363 delta_: f64,
8364 );
8365 pub fn atg_huber_loss_backward(
8366 out__: *mut *mut C_tensor,
8367 grad_output_: *mut C_tensor,
8368 self_: *mut C_tensor,
8369 target_: *mut C_tensor,
8370 reduction_: i64,
8371 delta_: f64,
8372 );
8373 pub fn atg_huber_loss_backward_out(
8374 out__: *mut *mut C_tensor,
8375 grad_input_: *mut C_tensor,
8376 grad_output_: *mut C_tensor,
8377 self_: *mut C_tensor,
8378 target_: *mut C_tensor,
8379 reduction_: i64,
8380 delta_: f64,
8381 );
8382 pub fn atg_huber_loss_out(
8383 out__: *mut *mut C_tensor,
8384 out_: *mut C_tensor,
8385 self_: *mut C_tensor,
8386 target_: *mut C_tensor,
8387 reduction_: i64,
8388 delta_: f64,
8389 );
8390 pub fn atg_hypot(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8391 pub fn atg_hypot_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8392 pub fn atg_hypot_out(
8393 out__: *mut *mut C_tensor,
8394 out_: *mut C_tensor,
8395 self_: *mut C_tensor,
8396 other_: *mut C_tensor,
8397 );
8398 pub fn atg_i0(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8399 pub fn atg_i0_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8400 pub fn atg_i0_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8401 pub fn atg_igamma(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8402 pub fn atg_igamma_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8403 pub fn atg_igamma_out(
8404 out__: *mut *mut C_tensor,
8405 out_: *mut C_tensor,
8406 self_: *mut C_tensor,
8407 other_: *mut C_tensor,
8408 );
8409 pub fn atg_igammac(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8410 pub fn atg_igammac_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8411 pub fn atg_igammac_out(
8412 out__: *mut *mut C_tensor,
8413 out_: *mut C_tensor,
8414 self_: *mut C_tensor,
8415 other_: *mut C_tensor,
8416 );
8417 pub fn atg_im2col(
8418 out__: *mut *mut C_tensor,
8419 self_: *mut C_tensor,
8420 kernel_size_data: *const i64,
8421 kernel_size_len: c_int,
8422 dilation_data: *const i64,
8423 dilation_len: c_int,
8424 padding_data: *const i64,
8425 padding_len: c_int,
8426 stride_data: *const i64,
8427 stride_len: c_int,
8428 );
8429 pub fn atg_im2col_out(
8430 out__: *mut *mut C_tensor,
8431 out_: *mut C_tensor,
8432 self_: *mut C_tensor,
8433 kernel_size_data: *const i64,
8434 kernel_size_len: c_int,
8435 dilation_data: *const i64,
8436 dilation_len: c_int,
8437 padding_data: *const i64,
8438 padding_len: c_int,
8439 stride_data: *const i64,
8440 stride_len: c_int,
8441 );
8442 pub fn atg_imag(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8443 pub fn atg_index(
8444 out__: *mut *mut C_tensor,
8445 self_: *mut C_tensor,
8446 indices_data: *const *mut C_tensor,
8447 indices_len: c_int,
8448 );
8449 pub fn atg_index_add(
8450 out__: *mut *mut C_tensor,
8451 self_: *mut C_tensor,
8452 dim_: i64,
8453 index_: *mut C_tensor,
8454 source_: *mut C_tensor,
8455 );
8456 pub fn atg_index_add_(
8457 out__: *mut *mut C_tensor,
8458 self_: *mut C_tensor,
8459 dim_: i64,
8460 index_: *mut C_tensor,
8461 source_: *mut C_tensor,
8462 );
8463 pub fn atg_index_add_out(
8464 out__: *mut *mut C_tensor,
8465 out_: *mut C_tensor,
8466 self_: *mut C_tensor,
8467 dim_: i64,
8468 index_: *mut C_tensor,
8469 source_: *mut C_tensor,
8470 );
8471 pub fn atg_index_copy(
8472 out__: *mut *mut C_tensor,
8473 self_: *mut C_tensor,
8474 dim_: i64,
8475 index_: *mut C_tensor,
8476 source_: *mut C_tensor,
8477 );
8478 pub fn atg_index_copy_(
8479 out__: *mut *mut C_tensor,
8480 self_: *mut C_tensor,
8481 dim_: i64,
8482 index_: *mut C_tensor,
8483 source_: *mut C_tensor,
8484 );
8485 pub fn atg_index_copy_out(
8486 out__: *mut *mut C_tensor,
8487 out_: *mut C_tensor,
8488 self_: *mut C_tensor,
8489 dim_: i64,
8490 index_: *mut C_tensor,
8491 source_: *mut C_tensor,
8492 );
8493 pub fn atg_index_fill(
8494 out__: *mut *mut C_tensor,
8495 self_: *mut C_tensor,
8496 dim_: i64,
8497 index_: *mut C_tensor,
8498 value_: *mut C_scalar,
8499 );
8500 pub fn atg_index_fill_(
8501 out__: *mut *mut C_tensor,
8502 self_: *mut C_tensor,
8503 dim_: i64,
8504 index_: *mut C_tensor,
8505 value_: *mut C_scalar,
8506 );
8507 pub fn atg_index_fill_int_scalar_out(
8508 out__: *mut *mut C_tensor,
8509 out_: *mut C_tensor,
8510 self_: *mut C_tensor,
8511 dim_: i64,
8512 index_: *mut C_tensor,
8513 value_: *mut C_scalar,
8514 );
8515 pub fn atg_index_fill_int_tensor(
8516 out__: *mut *mut C_tensor,
8517 self_: *mut C_tensor,
8518 dim_: i64,
8519 index_: *mut C_tensor,
8520 value_: *mut C_tensor,
8521 );
8522 pub fn atg_index_fill_int_tensor_(
8523 out__: *mut *mut C_tensor,
8524 self_: *mut C_tensor,
8525 dim_: i64,
8526 index_: *mut C_tensor,
8527 value_: *mut C_tensor,
8528 );
8529 pub fn atg_index_fill_int_tensor_out(
8530 out__: *mut *mut C_tensor,
8531 out_: *mut C_tensor,
8532 self_: *mut C_tensor,
8533 dim_: i64,
8534 index_: *mut C_tensor,
8535 value_: *mut C_tensor,
8536 );
8537 pub fn atg_index_put(
8538 out__: *mut *mut C_tensor,
8539 self_: *mut C_tensor,
8540 indices_data: *const *mut C_tensor,
8541 indices_len: c_int,
8542 values_: *mut C_tensor,
8543 accumulate_: c_int,
8544 );
8545 pub fn atg_index_put_(
8546 out__: *mut *mut C_tensor,
8547 self_: *mut C_tensor,
8548 indices_data: *const *mut C_tensor,
8549 indices_len: c_int,
8550 values_: *mut C_tensor,
8551 accumulate_: c_int,
8552 );
8553 pub fn atg_index_put_out(
8554 out__: *mut *mut C_tensor,
8555 out_: *mut C_tensor,
8556 self_: *mut C_tensor,
8557 indices_data: *const *mut C_tensor,
8558 indices_len: c_int,
8559 values_: *mut C_tensor,
8560 accumulate_: c_int,
8561 );
8562 pub fn atg_index_reduce(
8563 out__: *mut *mut C_tensor,
8564 self_: *mut C_tensor,
8565 dim_: i64,
8566 index_: *mut C_tensor,
8567 source_: *mut C_tensor,
8568 reduce_ptr: *const u8,
8569 reduce_len: c_int,
8570 include_self_: c_int,
8571 );
8572 pub fn atg_index_reduce_(
8573 out__: *mut *mut C_tensor,
8574 self_: *mut C_tensor,
8575 dim_: i64,
8576 index_: *mut C_tensor,
8577 source_: *mut C_tensor,
8578 reduce_ptr: *const u8,
8579 reduce_len: c_int,
8580 include_self_: c_int,
8581 );
8582 pub fn atg_index_reduce_out(
8583 out__: *mut *mut C_tensor,
8584 out_: *mut C_tensor,
8585 self_: *mut C_tensor,
8586 dim_: i64,
8587 index_: *mut C_tensor,
8588 source_: *mut C_tensor,
8589 reduce_ptr: *const u8,
8590 reduce_len: c_int,
8591 include_self_: c_int,
8592 );
8593 pub fn atg_index_select(
8594 out__: *mut *mut C_tensor,
8595 self_: *mut C_tensor,
8596 dim_: i64,
8597 index_: *mut C_tensor,
8598 );
8599 pub fn atg_index_select_backward(
8600 out__: *mut *mut C_tensor,
8601 grad_: *mut C_tensor,
8602 self_sizes_data: *const i64,
8603 self_sizes_len: c_int,
8604 dim_: i64,
8605 index_: *mut C_tensor,
8606 );
8607 pub fn atg_index_select_out(
8608 out__: *mut *mut C_tensor,
8609 out_: *mut C_tensor,
8610 self_: *mut C_tensor,
8611 dim_: i64,
8612 index_: *mut C_tensor,
8613 );
8614 pub fn atg_index_tensor_out(
8615 out__: *mut *mut C_tensor,
8616 out_: *mut C_tensor,
8617 self_: *mut C_tensor,
8618 indices_data: *const *mut C_tensor,
8619 indices_len: c_int,
8620 );
8621 pub fn atg_indices(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8622 pub fn atg_indices_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8623 pub fn atg_indices_copy_out(
8624 out__: *mut *mut C_tensor,
8625 out_: *mut C_tensor,
8626 self_: *mut C_tensor,
8627 );
8628 pub fn atg_infinitely_differentiable_gelu_backward(
8629 out__: *mut *mut C_tensor,
8630 grad_: *mut C_tensor,
8631 self_: *mut C_tensor,
8632 );
8633 pub fn atg_inner(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8634 pub fn atg_inner_out(
8635 out__: *mut *mut C_tensor,
8636 out_: *mut C_tensor,
8637 self_: *mut C_tensor,
8638 other_: *mut C_tensor,
8639 );
8640 pub fn atg_instance_norm(
8641 out__: *mut *mut C_tensor,
8642 input_: *mut C_tensor,
8643 weight_: *mut C_tensor,
8644 bias_: *mut C_tensor,
8645 running_mean_: *mut C_tensor,
8646 running_var_: *mut C_tensor,
8647 use_input_stats_: c_int,
8648 momentum_: f64,
8649 eps_: f64,
8650 cudnn_enabled_: c_int,
8651 );
8652 pub fn atg_int_repr(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8653 pub fn atg_int_repr_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8654 pub fn atg_inverse(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8655 pub fn atg_inverse_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8656 pub fn atg_is_coalesced(self_: *mut C_tensor) -> c_int;
8657 pub fn atg_is_complex(self_: *mut C_tensor) -> c_int;
8658 pub fn atg_is_conj(self_: *mut C_tensor) -> c_int;
8659 pub fn atg_is_distributed(self_: *mut C_tensor) -> c_int;
8660 pub fn atg_is_floating_point(self_: *mut C_tensor) -> c_int;
8661 pub fn atg_is_inference(self_: *mut C_tensor) -> c_int;
8662 pub fn atg_is_leaf(self_: *mut C_tensor) -> c_int;
8663 pub fn atg_is_neg(self_: *mut C_tensor) -> c_int;
8664 pub fn atg_is_nonzero(self_: *mut C_tensor) -> c_int;
8665 pub fn atg_is_pinned(self_: *mut C_tensor, device_: c_int) -> c_int;
8666 pub fn atg_is_same_size(self_: *mut C_tensor, other_: *mut C_tensor) -> c_int;
8667 pub fn atg_is_set_to(self_: *mut C_tensor, tensor_: *mut C_tensor) -> c_int;
8668 pub fn atg_is_signed(self_: *mut C_tensor) -> c_int;
8669 pub fn atg_is_vulkan_available() -> c_int;
8670 pub fn atg_isclose(
8671 out__: *mut *mut C_tensor,
8672 self_: *mut C_tensor,
8673 other_: *mut C_tensor,
8674 rtol_: f64,
8675 atol_: f64,
8676 equal_nan_: c_int,
8677 );
8678 pub fn atg_isfinite(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8679 pub fn atg_isin(
8680 out__: *mut *mut C_tensor,
8681 elements_: *mut C_tensor,
8682 test_elements_: *mut C_tensor,
8683 assume_unique_: c_int,
8684 invert_: c_int,
8685 );
8686 pub fn atg_isin_scalar_tensor(
8687 out__: *mut *mut C_tensor,
8688 element_: *mut C_scalar,
8689 test_elements_: *mut C_tensor,
8690 assume_unique_: c_int,
8691 invert_: c_int,
8692 );
8693 pub fn atg_isin_scalar_tensor_out(
8694 out__: *mut *mut C_tensor,
8695 out_: *mut C_tensor,
8696 element_: *mut C_scalar,
8697 test_elements_: *mut C_tensor,
8698 assume_unique_: c_int,
8699 invert_: c_int,
8700 );
8701 pub fn atg_isin_tensor_scalar(
8702 out__: *mut *mut C_tensor,
8703 elements_: *mut C_tensor,
8704 test_element_: *mut C_scalar,
8705 assume_unique_: c_int,
8706 invert_: c_int,
8707 );
8708 pub fn atg_isin_tensor_scalar_out(
8709 out__: *mut *mut C_tensor,
8710 out_: *mut C_tensor,
8711 elements_: *mut C_tensor,
8712 test_element_: *mut C_scalar,
8713 assume_unique_: c_int,
8714 invert_: c_int,
8715 );
8716 pub fn atg_isin_tensor_tensor_out(
8717 out__: *mut *mut C_tensor,
8718 out_: *mut C_tensor,
8719 elements_: *mut C_tensor,
8720 test_elements_: *mut C_tensor,
8721 assume_unique_: c_int,
8722 invert_: c_int,
8723 );
8724 pub fn atg_isinf(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8725 pub fn atg_isinf_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8726 pub fn atg_isnan(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8727 pub fn atg_isnan_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8728 pub fn atg_isneginf(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8729 pub fn atg_isneginf_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8730 pub fn atg_isposinf(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8731 pub fn atg_isposinf_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8732 pub fn atg_isreal(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8733 pub fn atg_istft(
8734 out__: *mut *mut C_tensor,
8735 self_: *mut C_tensor,
8736 n_fft_: i64,
8737 hop_length_v: i64,
8738 hop_length_null: i8,
8739 win_length_v: i64,
8740 win_length_null: i8,
8741 window_: *mut C_tensor,
8742 center_: c_int,
8743 normalized_: c_int,
8744 onesided_: c_int,
8745 length_v: i64,
8746 length_null: i8,
8747 return_complex_: c_int,
8748 );
8749 pub fn atg_kaiser_window(
8750 out__: *mut *mut C_tensor,
8751 window_length_: i64,
8752 options_kind: c_int,
8753 options_device: c_int,
8754 );
8755 pub fn atg_kaiser_window_beta(
8756 out__: *mut *mut C_tensor,
8757 window_length_: i64,
8758 periodic_: c_int,
8759 beta_: f64,
8760 options_kind: c_int,
8761 options_device: c_int,
8762 );
8763 pub fn atg_kaiser_window_beta_out(
8764 out__: *mut *mut C_tensor,
8765 out_: *mut C_tensor,
8766 window_length_: i64,
8767 periodic_: c_int,
8768 beta_: f64,
8769 );
8770 pub fn atg_kaiser_window_out(
8771 out__: *mut *mut C_tensor,
8772 out_: *mut C_tensor,
8773 window_length_: i64,
8774 );
8775 pub fn atg_kaiser_window_periodic(
8776 out__: *mut *mut C_tensor,
8777 window_length_: i64,
8778 periodic_: c_int,
8779 options_kind: c_int,
8780 options_device: c_int,
8781 );
8782 pub fn atg_kaiser_window_periodic_out(
8783 out__: *mut *mut C_tensor,
8784 out_: *mut C_tensor,
8785 window_length_: i64,
8786 periodic_: c_int,
8787 );
8788 pub fn atg_kl_div(
8789 out__: *mut *mut C_tensor,
8790 self_: *mut C_tensor,
8791 target_: *mut C_tensor,
8792 reduction_: i64,
8793 log_target_: c_int,
8794 );
8795 pub fn atg_kron(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8796 pub fn atg_kron_out(
8797 out__: *mut *mut C_tensor,
8798 out_: *mut C_tensor,
8799 self_: *mut C_tensor,
8800 other_: *mut C_tensor,
8801 );
8802 pub fn atg_kthvalue(
8803 out__: *mut *mut C_tensor,
8804 self_: *mut C_tensor,
8805 k_: i64,
8806 dim_: i64,
8807 keepdim_: c_int,
8808 );
8809 pub fn atg_kthvalue_values(
8810 out__: *mut *mut C_tensor,
8811 values_: *mut C_tensor,
8812 indices_: *mut C_tensor,
8813 self_: *mut C_tensor,
8814 k_: i64,
8815 dim_: i64,
8816 keepdim_: c_int,
8817 );
8818 pub fn atg_l1_loss(
8819 out__: *mut *mut C_tensor,
8820 self_: *mut C_tensor,
8821 target_: *mut C_tensor,
8822 reduction_: i64,
8823 );
8824 pub fn atg_layer_norm(
8825 out__: *mut *mut C_tensor,
8826 input_: *mut C_tensor,
8827 normalized_shape_data: *const i64,
8828 normalized_shape_len: c_int,
8829 weight_: *mut C_tensor,
8830 bias_: *mut C_tensor,
8831 eps_: f64,
8832 cudnn_enable_: c_int,
8833 );
8834 pub fn atg_lcm(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8835 pub fn atg_lcm_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8836 pub fn atg_lcm_out(
8837 out__: *mut *mut C_tensor,
8838 out_: *mut C_tensor,
8839 self_: *mut C_tensor,
8840 other_: *mut C_tensor,
8841 );
8842 pub fn atg_ldexp(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8843 pub fn atg_ldexp_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8844 pub fn atg_ldexp_out(
8845 out__: *mut *mut C_tensor,
8846 out_: *mut C_tensor,
8847 self_: *mut C_tensor,
8848 other_: *mut C_tensor,
8849 );
8850 pub fn atg_le(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
8851 pub fn atg_le_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
8852 pub fn atg_le_scalar_out(
8853 out__: *mut *mut C_tensor,
8854 out_: *mut C_tensor,
8855 self_: *mut C_tensor,
8856 other_: *mut C_scalar,
8857 );
8858 pub fn atg_le_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8859 pub fn atg_le_tensor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8860 pub fn atg_le_tensor_out(
8861 out__: *mut *mut C_tensor,
8862 out_: *mut C_tensor,
8863 self_: *mut C_tensor,
8864 other_: *mut C_tensor,
8865 );
8866 pub fn atg_leaky_relu(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8867 pub fn atg_leaky_relu_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8868 pub fn atg_leaky_relu_backward(
8869 out__: *mut *mut C_tensor,
8870 grad_output_: *mut C_tensor,
8871 self_: *mut C_tensor,
8872 negative_slope_: *mut C_scalar,
8873 self_is_result_: c_int,
8874 );
8875 pub fn atg_leaky_relu_backward_grad_input(
8876 out__: *mut *mut C_tensor,
8877 grad_input_: *mut C_tensor,
8878 grad_output_: *mut C_tensor,
8879 self_: *mut C_tensor,
8880 negative_slope_: *mut C_scalar,
8881 self_is_result_: c_int,
8882 );
8883 pub fn atg_leaky_relu_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8884 pub fn atg_lerp(
8885 out__: *mut *mut C_tensor,
8886 self_: *mut C_tensor,
8887 end_: *mut C_tensor,
8888 weight_: *mut C_scalar,
8889 );
8890 pub fn atg_lerp_(
8891 out__: *mut *mut C_tensor,
8892 self_: *mut C_tensor,
8893 end_: *mut C_tensor,
8894 weight_: *mut C_scalar,
8895 );
8896 pub fn atg_lerp_scalar_out(
8897 out__: *mut *mut C_tensor,
8898 out_: *mut C_tensor,
8899 self_: *mut C_tensor,
8900 end_: *mut C_tensor,
8901 weight_: *mut C_scalar,
8902 );
8903 pub fn atg_lerp_tensor(
8904 out__: *mut *mut C_tensor,
8905 self_: *mut C_tensor,
8906 end_: *mut C_tensor,
8907 weight_: *mut C_tensor,
8908 );
8909 pub fn atg_lerp_tensor_(
8910 out__: *mut *mut C_tensor,
8911 self_: *mut C_tensor,
8912 end_: *mut C_tensor,
8913 weight_: *mut C_tensor,
8914 );
8915 pub fn atg_lerp_tensor_out(
8916 out__: *mut *mut C_tensor,
8917 out_: *mut C_tensor,
8918 self_: *mut C_tensor,
8919 end_: *mut C_tensor,
8920 weight_: *mut C_tensor,
8921 );
8922 pub fn atg_less(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
8923 pub fn atg_less_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
8924 pub fn atg_less_equal(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
8925 pub fn atg_less_equal_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
8926 pub fn atg_less_equal_scalar_out(
8927 out__: *mut *mut C_tensor,
8928 out_: *mut C_tensor,
8929 self_: *mut C_tensor,
8930 other_: *mut C_scalar,
8931 );
8932 pub fn atg_less_equal_tensor(
8933 out__: *mut *mut C_tensor,
8934 self_: *mut C_tensor,
8935 other_: *mut C_tensor,
8936 );
8937 pub fn atg_less_equal_tensor_(
8938 out__: *mut *mut C_tensor,
8939 self_: *mut C_tensor,
8940 other_: *mut C_tensor,
8941 );
8942 pub fn atg_less_equal_tensor_out(
8943 out__: *mut *mut C_tensor,
8944 out_: *mut C_tensor,
8945 self_: *mut C_tensor,
8946 other_: *mut C_tensor,
8947 );
8948 pub fn atg_less_scalar_out(
8949 out__: *mut *mut C_tensor,
8950 out_: *mut C_tensor,
8951 self_: *mut C_tensor,
8952 other_: *mut C_scalar,
8953 );
8954 pub fn atg_less_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8955 pub fn atg_less_tensor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
8956 pub fn atg_less_tensor_out(
8957 out__: *mut *mut C_tensor,
8958 out_: *mut C_tensor,
8959 self_: *mut C_tensor,
8960 other_: *mut C_tensor,
8961 );
8962 pub fn atg_lgamma(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8963 pub fn atg_lgamma_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8964 pub fn atg_lgamma_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8965 pub fn atg_lift(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8966 pub fn atg_lift_fresh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8967 pub fn atg_lift_fresh_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
8968 pub fn atg_lift_fresh_copy_out(
8969 out__: *mut *mut C_tensor,
8970 out_: *mut C_tensor,
8971 self_: *mut C_tensor,
8972 );
8973 pub fn atg_lift_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
8974 pub fn atg_linalg_cholesky(out__: *mut *mut C_tensor, self_: *mut C_tensor, upper_: c_int);
8975 pub fn atg_linalg_cholesky_ex(
8976 out__: *mut *mut C_tensor,
8977 self_: *mut C_tensor,
8978 upper_: c_int,
8979 check_errors_: c_int,
8980 );
8981 pub fn atg_linalg_cholesky_ex_l(
8982 out__: *mut *mut C_tensor,
8983 L_: *mut C_tensor,
8984 info_: *mut C_tensor,
8985 self_: *mut C_tensor,
8986 upper_: c_int,
8987 check_errors_: c_int,
8988 );
8989 pub fn atg_linalg_cholesky_out(
8990 out__: *mut *mut C_tensor,
8991 out_: *mut C_tensor,
8992 self_: *mut C_tensor,
8993 upper_: c_int,
8994 );
8995 pub fn atg_linalg_cond(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: *mut C_scalar);
8996 pub fn atg_linalg_cond_out(
8997 out__: *mut *mut C_tensor,
8998 out_: *mut C_tensor,
8999 self_: *mut C_tensor,
9000 p_: *mut C_scalar,
9001 );
9002 pub fn atg_linalg_cond_p_str(
9003 out__: *mut *mut C_tensor,
9004 self_: *mut C_tensor,
9005 p_ptr: *const u8,
9006 p_len: c_int,
9007 );
9008 pub fn atg_linalg_cond_p_str_out(
9009 out__: *mut *mut C_tensor,
9010 out_: *mut C_tensor,
9011 self_: *mut C_tensor,
9012 p_ptr: *const u8,
9013 p_len: c_int,
9014 );
9015 pub fn atg_linalg_cross(
9016 out__: *mut *mut C_tensor,
9017 self_: *mut C_tensor,
9018 other_: *mut C_tensor,
9019 dim_: i64,
9020 );
9021 pub fn atg_linalg_cross_out(
9022 out__: *mut *mut C_tensor,
9023 out_: *mut C_tensor,
9024 self_: *mut C_tensor,
9025 other_: *mut C_tensor,
9026 dim_: i64,
9027 );
9028 pub fn atg_linalg_det(out__: *mut *mut C_tensor, A_: *mut C_tensor);
9029 pub fn atg_linalg_det_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, A_: *mut C_tensor);
9030 pub fn atg_linalg_diagonal(
9031 out__: *mut *mut C_tensor,
9032 A_: *mut C_tensor,
9033 offset_: i64,
9034 dim1_: i64,
9035 dim2_: i64,
9036 );
9037 pub fn atg_linalg_eig(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9038 pub fn atg_linalg_eig_out(
9039 out__: *mut *mut C_tensor,
9040 eigenvalues_: *mut C_tensor,
9041 eigenvectors_: *mut C_tensor,
9042 self_: *mut C_tensor,
9043 );
9044 pub fn atg_linalg_eigh(
9045 out__: *mut *mut C_tensor,
9046 self_: *mut C_tensor,
9047 UPLO_ptr: *const u8,
9048 UPLO_len: c_int,
9049 );
9050 pub fn atg_linalg_eigh_eigvals(
9051 out__: *mut *mut C_tensor,
9052 eigvals_: *mut C_tensor,
9053 eigvecs_: *mut C_tensor,
9054 self_: *mut C_tensor,
9055 UPLO_ptr: *const u8,
9056 UPLO_len: c_int,
9057 );
9058 pub fn atg_linalg_eigvals(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9059 pub fn atg_linalg_eigvals_out(
9060 out__: *mut *mut C_tensor,
9061 out_: *mut C_tensor,
9062 self_: *mut C_tensor,
9063 );
9064 pub fn atg_linalg_eigvalsh(
9065 out__: *mut *mut C_tensor,
9066 self_: *mut C_tensor,
9067 UPLO_ptr: *const u8,
9068 UPLO_len: c_int,
9069 );
9070 pub fn atg_linalg_eigvalsh_out(
9071 out__: *mut *mut C_tensor,
9072 out_: *mut C_tensor,
9073 self_: *mut C_tensor,
9074 UPLO_ptr: *const u8,
9075 UPLO_len: c_int,
9076 );
9077 pub fn atg_linalg_householder_product(
9078 out__: *mut *mut C_tensor,
9079 input_: *mut C_tensor,
9080 tau_: *mut C_tensor,
9081 );
9082 pub fn atg_linalg_householder_product_out(
9083 out__: *mut *mut C_tensor,
9084 out_: *mut C_tensor,
9085 input_: *mut C_tensor,
9086 tau_: *mut C_tensor,
9087 );
9088 pub fn atg_linalg_inv(out__: *mut *mut C_tensor, A_: *mut C_tensor);
9089 pub fn atg_linalg_inv_ex(out__: *mut *mut C_tensor, A_: *mut C_tensor, check_errors_: c_int);
9090 pub fn atg_linalg_inv_ex_inverse(
9091 out__: *mut *mut C_tensor,
9092 inverse_: *mut C_tensor,
9093 info_: *mut C_tensor,
9094 A_: *mut C_tensor,
9095 check_errors_: c_int,
9096 );
9097 pub fn atg_linalg_inv_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, A_: *mut C_tensor);
9098 pub fn atg_linalg_ldl_factor(
9099 out__: *mut *mut C_tensor,
9100 self_: *mut C_tensor,
9101 hermitian_: c_int,
9102 );
9103 pub fn atg_linalg_ldl_factor_ex(
9104 out__: *mut *mut C_tensor,
9105 self_: *mut C_tensor,
9106 hermitian_: c_int,
9107 check_errors_: c_int,
9108 );
9109 pub fn atg_linalg_ldl_factor_ex_out(
9110 out__: *mut *mut C_tensor,
9111 LD_: *mut C_tensor,
9112 pivots_: *mut C_tensor,
9113 info_: *mut C_tensor,
9114 self_: *mut C_tensor,
9115 hermitian_: c_int,
9116 check_errors_: c_int,
9117 );
9118 pub fn atg_linalg_ldl_factor_out(
9119 out__: *mut *mut C_tensor,
9120 LD_: *mut C_tensor,
9121 pivots_: *mut C_tensor,
9122 self_: *mut C_tensor,
9123 hermitian_: c_int,
9124 );
9125 pub fn atg_linalg_ldl_solve(
9126 out__: *mut *mut C_tensor,
9127 LD_: *mut C_tensor,
9128 pivots_: *mut C_tensor,
9129 B_: *mut C_tensor,
9130 hermitian_: c_int,
9131 );
9132 pub fn atg_linalg_ldl_solve_out(
9133 out__: *mut *mut C_tensor,
9134 out_: *mut C_tensor,
9135 LD_: *mut C_tensor,
9136 pivots_: *mut C_tensor,
9137 B_: *mut C_tensor,
9138 hermitian_: c_int,
9139 );
9140 pub fn atg_linalg_lstsq(
9141 out__: *mut *mut C_tensor,
9142 self_: *mut C_tensor,
9143 b_: *mut C_tensor,
9144 rcond_v: f64,
9145 rcond_null: i8,
9146 driver_ptr: *const u8,
9147 driver_len: c_int,
9148 );
9149 pub fn atg_linalg_lstsq_out(
9150 out__: *mut *mut C_tensor,
9151 solution_: *mut C_tensor,
9152 residuals_: *mut C_tensor,
9153 rank_: *mut C_tensor,
9154 singular_values_: *mut C_tensor,
9155 self_: *mut C_tensor,
9156 b_: *mut C_tensor,
9157 rcond_v: f64,
9158 rcond_null: i8,
9159 driver_ptr: *const u8,
9160 driver_len: c_int,
9161 );
9162 pub fn atg_linalg_lu(out__: *mut *mut C_tensor, A_: *mut C_tensor, pivot_: c_int);
9163 pub fn atg_linalg_lu_factor(out__: *mut *mut C_tensor, A_: *mut C_tensor, pivot_: c_int);
9164 pub fn atg_linalg_lu_factor_ex(
9165 out__: *mut *mut C_tensor,
9166 A_: *mut C_tensor,
9167 pivot_: c_int,
9168 check_errors_: c_int,
9169 );
9170 pub fn atg_linalg_lu_factor_ex_out(
9171 out__: *mut *mut C_tensor,
9172 LU_: *mut C_tensor,
9173 pivots_: *mut C_tensor,
9174 info_: *mut C_tensor,
9175 A_: *mut C_tensor,
9176 pivot_: c_int,
9177 check_errors_: c_int,
9178 );
9179 pub fn atg_linalg_lu_factor_out(
9180 out__: *mut *mut C_tensor,
9181 LU_: *mut C_tensor,
9182 pivots_: *mut C_tensor,
9183 A_: *mut C_tensor,
9184 pivot_: c_int,
9185 );
9186 pub fn atg_linalg_lu_out(
9187 out__: *mut *mut C_tensor,
9188 P_: *mut C_tensor,
9189 L_: *mut C_tensor,
9190 U_: *mut C_tensor,
9191 A_: *mut C_tensor,
9192 pivot_: c_int,
9193 );
9194 pub fn atg_linalg_lu_solve(
9195 out__: *mut *mut C_tensor,
9196 LU_: *mut C_tensor,
9197 pivots_: *mut C_tensor,
9198 B_: *mut C_tensor,
9199 left_: c_int,
9200 adjoint_: c_int,
9201 );
9202 pub fn atg_linalg_lu_solve_out(
9203 out__: *mut *mut C_tensor,
9204 out_: *mut C_tensor,
9205 LU_: *mut C_tensor,
9206 pivots_: *mut C_tensor,
9207 B_: *mut C_tensor,
9208 left_: c_int,
9209 adjoint_: c_int,
9210 );
9211 pub fn atg_linalg_matmul(
9212 out__: *mut *mut C_tensor,
9213 self_: *mut C_tensor,
9214 other_: *mut C_tensor,
9215 );
9216 pub fn atg_linalg_matmul_out(
9217 out__: *mut *mut C_tensor,
9218 out_: *mut C_tensor,
9219 self_: *mut C_tensor,
9220 other_: *mut C_tensor,
9221 );
9222 pub fn atg_linalg_matrix_exp(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9223 pub fn atg_linalg_matrix_exp_out(
9224 out__: *mut *mut C_tensor,
9225 out_: *mut C_tensor,
9226 self_: *mut C_tensor,
9227 );
9228 pub fn atg_linalg_matrix_power(out__: *mut *mut C_tensor, self_: *mut C_tensor, n_: i64);
9229 pub fn atg_linalg_matrix_power_out(
9230 out__: *mut *mut C_tensor,
9231 out_: *mut C_tensor,
9232 self_: *mut C_tensor,
9233 n_: i64,
9234 );
9235 pub fn atg_linalg_matrix_rank(
9236 out__: *mut *mut C_tensor,
9237 self_: *mut C_tensor,
9238 tol_: f64,
9239 hermitian_: c_int,
9240 );
9241 pub fn atg_linalg_matrix_rank_atol_rtol_float(
9242 out__: *mut *mut C_tensor,
9243 self_: *mut C_tensor,
9244 atol_v: f64,
9245 atol_null: i8,
9246 rtol_v: f64,
9247 rtol_null: i8,
9248 hermitian_: c_int,
9249 );
9250 pub fn atg_linalg_matrix_rank_atol_rtol_float_out(
9251 out__: *mut *mut C_tensor,
9252 out_: *mut C_tensor,
9253 self_: *mut C_tensor,
9254 atol_v: f64,
9255 atol_null: i8,
9256 rtol_v: f64,
9257 rtol_null: i8,
9258 hermitian_: c_int,
9259 );
9260 pub fn atg_linalg_matrix_rank_atol_rtol_tensor(
9261 out__: *mut *mut C_tensor,
9262 input_: *mut C_tensor,
9263 atol_: *mut C_tensor,
9264 rtol_: *mut C_tensor,
9265 hermitian_: c_int,
9266 );
9267 pub fn atg_linalg_matrix_rank_atol_rtol_tensor_out(
9268 out__: *mut *mut C_tensor,
9269 out_: *mut C_tensor,
9270 input_: *mut C_tensor,
9271 atol_: *mut C_tensor,
9272 rtol_: *mut C_tensor,
9273 hermitian_: c_int,
9274 );
9275 pub fn atg_linalg_matrix_rank_out(
9276 out__: *mut *mut C_tensor,
9277 out_: *mut C_tensor,
9278 self_: *mut C_tensor,
9279 tol_: f64,
9280 hermitian_: c_int,
9281 );
9282 pub fn atg_linalg_matrix_rank_out_tol_tensor(
9283 out__: *mut *mut C_tensor,
9284 out_: *mut C_tensor,
9285 input_: *mut C_tensor,
9286 tol_: *mut C_tensor,
9287 hermitian_: c_int,
9288 );
9289 pub fn atg_linalg_matrix_rank_tol_tensor(
9290 out__: *mut *mut C_tensor,
9291 input_: *mut C_tensor,
9292 tol_: *mut C_tensor,
9293 hermitian_: c_int,
9294 );
9295 pub fn atg_linalg_multi_dot(
9296 out__: *mut *mut C_tensor,
9297 tensors_data: *const *mut C_tensor,
9298 tensors_len: c_int,
9299 );
9300 pub fn atg_linalg_multi_dot_out(
9301 out__: *mut *mut C_tensor,
9302 out_: *mut C_tensor,
9303 tensors_data: *const *mut C_tensor,
9304 tensors_len: c_int,
9305 );
9306 pub fn atg_linalg_norm(
9307 out__: *mut *mut C_tensor,
9308 self_: *mut C_tensor,
9309 ord_: *mut C_scalar,
9310 dim_data: *const i64,
9311 dim_len: c_int,
9312 keepdim_: c_int,
9313 dtype_: c_int,
9314 );
9315 pub fn atg_linalg_norm_ord_str(
9316 out__: *mut *mut C_tensor,
9317 self_: *mut C_tensor,
9318 ord_ptr: *const u8,
9319 ord_len: c_int,
9320 dim_data: *const i64,
9321 dim_len: c_int,
9322 keepdim_: c_int,
9323 dtype_: c_int,
9324 );
9325 pub fn atg_linalg_norm_ord_str_out(
9326 out__: *mut *mut C_tensor,
9327 out_: *mut C_tensor,
9328 self_: *mut C_tensor,
9329 ord_ptr: *const u8,
9330 ord_len: c_int,
9331 dim_data: *const i64,
9332 dim_len: c_int,
9333 keepdim_: c_int,
9334 dtype_: c_int,
9335 );
9336 pub fn atg_linalg_norm_out(
9337 out__: *mut *mut C_tensor,
9338 out_: *mut C_tensor,
9339 self_: *mut C_tensor,
9340 ord_: *mut C_scalar,
9341 dim_data: *const i64,
9342 dim_len: c_int,
9343 keepdim_: c_int,
9344 dtype_: c_int,
9345 );
9346 pub fn atg_linalg_pinv(
9347 out__: *mut *mut C_tensor,
9348 self_: *mut C_tensor,
9349 rcond_: f64,
9350 hermitian_: c_int,
9351 );
9352 pub fn atg_linalg_pinv_atol_rtol_float(
9353 out__: *mut *mut C_tensor,
9354 self_: *mut C_tensor,
9355 atol_v: f64,
9356 atol_null: i8,
9357 rtol_v: f64,
9358 rtol_null: i8,
9359 hermitian_: c_int,
9360 );
9361 pub fn atg_linalg_pinv_atol_rtol_float_out(
9362 out__: *mut *mut C_tensor,
9363 out_: *mut C_tensor,
9364 self_: *mut C_tensor,
9365 atol_v: f64,
9366 atol_null: i8,
9367 rtol_v: f64,
9368 rtol_null: i8,
9369 hermitian_: c_int,
9370 );
9371 pub fn atg_linalg_pinv_atol_rtol_tensor(
9372 out__: *mut *mut C_tensor,
9373 self_: *mut C_tensor,
9374 atol_: *mut C_tensor,
9375 rtol_: *mut C_tensor,
9376 hermitian_: c_int,
9377 );
9378 pub fn atg_linalg_pinv_atol_rtol_tensor_out(
9379 out__: *mut *mut C_tensor,
9380 out_: *mut C_tensor,
9381 self_: *mut C_tensor,
9382 atol_: *mut C_tensor,
9383 rtol_: *mut C_tensor,
9384 hermitian_: c_int,
9385 );
9386 pub fn atg_linalg_pinv_out(
9387 out__: *mut *mut C_tensor,
9388 out_: *mut C_tensor,
9389 self_: *mut C_tensor,
9390 rcond_: f64,
9391 hermitian_: c_int,
9392 );
9393 pub fn atg_linalg_pinv_out_rcond_tensor(
9394 out__: *mut *mut C_tensor,
9395 out_: *mut C_tensor,
9396 self_: *mut C_tensor,
9397 rcond_: *mut C_tensor,
9398 hermitian_: c_int,
9399 );
9400 pub fn atg_linalg_pinv_rcond_tensor(
9401 out__: *mut *mut C_tensor,
9402 self_: *mut C_tensor,
9403 rcond_: *mut C_tensor,
9404 hermitian_: c_int,
9405 );
9406 pub fn atg_linalg_qr(
9407 out__: *mut *mut C_tensor,
9408 A_: *mut C_tensor,
9409 mode_ptr: *const u8,
9410 mode_len: c_int,
9411 );
9412 pub fn atg_linalg_qr_out(
9413 out__: *mut *mut C_tensor,
9414 Q_: *mut C_tensor,
9415 R_: *mut C_tensor,
9416 A_: *mut C_tensor,
9417 mode_ptr: *const u8,
9418 mode_len: c_int,
9419 );
9420 pub fn atg_linalg_slogdet(out__: *mut *mut C_tensor, A_: *mut C_tensor);
9421 pub fn atg_linalg_slogdet_out(
9422 out__: *mut *mut C_tensor,
9423 sign_: *mut C_tensor,
9424 logabsdet_: *mut C_tensor,
9425 A_: *mut C_tensor,
9426 );
9427 pub fn atg_linalg_solve(
9428 out__: *mut *mut C_tensor,
9429 A_: *mut C_tensor,
9430 B_: *mut C_tensor,
9431 left_: c_int,
9432 );
9433 pub fn atg_linalg_solve_ex(
9434 out__: *mut *mut C_tensor,
9435 A_: *mut C_tensor,
9436 B_: *mut C_tensor,
9437 left_: c_int,
9438 check_errors_: c_int,
9439 );
9440 pub fn atg_linalg_solve_ex_out(
9441 out__: *mut *mut C_tensor,
9442 result_: *mut C_tensor,
9443 info_: *mut C_tensor,
9444 A_: *mut C_tensor,
9445 B_: *mut C_tensor,
9446 left_: c_int,
9447 check_errors_: c_int,
9448 );
9449 pub fn atg_linalg_solve_out(
9450 out__: *mut *mut C_tensor,
9451 out_: *mut C_tensor,
9452 A_: *mut C_tensor,
9453 B_: *mut C_tensor,
9454 left_: c_int,
9455 );
9456 pub fn atg_linalg_solve_triangular(
9457 out__: *mut *mut C_tensor,
9458 self_: *mut C_tensor,
9459 B_: *mut C_tensor,
9460 upper_: c_int,
9461 left_: c_int,
9462 unitriangular_: c_int,
9463 );
9464 pub fn atg_linalg_solve_triangular_out(
9465 out__: *mut *mut C_tensor,
9466 out_: *mut C_tensor,
9467 self_: *mut C_tensor,
9468 B_: *mut C_tensor,
9469 upper_: c_int,
9470 left_: c_int,
9471 unitriangular_: c_int,
9472 );
9473 pub fn atg_linalg_svd(
9474 out__: *mut *mut C_tensor,
9475 A_: *mut C_tensor,
9476 full_matrices_: c_int,
9477 driver_ptr: *const u8,
9478 driver_len: c_int,
9479 );
9480 pub fn atg_linalg_svd_u(
9481 out__: *mut *mut C_tensor,
9482 U_: *mut C_tensor,
9483 S_: *mut C_tensor,
9484 Vh_: *mut C_tensor,
9485 A_: *mut C_tensor,
9486 full_matrices_: c_int,
9487 driver_ptr: *const u8,
9488 driver_len: c_int,
9489 );
9490 pub fn atg_linalg_svdvals(
9491 out__: *mut *mut C_tensor,
9492 A_: *mut C_tensor,
9493 driver_ptr: *const u8,
9494 driver_len: c_int,
9495 );
9496 pub fn atg_linalg_svdvals_out(
9497 out__: *mut *mut C_tensor,
9498 out_: *mut C_tensor,
9499 A_: *mut C_tensor,
9500 driver_ptr: *const u8,
9501 driver_len: c_int,
9502 );
9503 pub fn atg_linalg_tensorinv(out__: *mut *mut C_tensor, self_: *mut C_tensor, ind_: i64);
9504 pub fn atg_linalg_tensorinv_out(
9505 out__: *mut *mut C_tensor,
9506 out_: *mut C_tensor,
9507 self_: *mut C_tensor,
9508 ind_: i64,
9509 );
9510 pub fn atg_linalg_tensorsolve(
9511 out__: *mut *mut C_tensor,
9512 self_: *mut C_tensor,
9513 other_: *mut C_tensor,
9514 dims_data: *const i64,
9515 dims_len: c_int,
9516 );
9517 pub fn atg_linalg_tensorsolve_out(
9518 out__: *mut *mut C_tensor,
9519 out_: *mut C_tensor,
9520 self_: *mut C_tensor,
9521 other_: *mut C_tensor,
9522 dims_data: *const i64,
9523 dims_len: c_int,
9524 );
9525 pub fn atg_linalg_vander(out__: *mut *mut C_tensor, x_: *mut C_tensor, n_v: i64, n_null: i8);
9526 pub fn atg_linalg_vecdot(
9527 out__: *mut *mut C_tensor,
9528 x_: *mut C_tensor,
9529 y_: *mut C_tensor,
9530 dim_: i64,
9531 );
9532 pub fn atg_linalg_vecdot_out(
9533 out__: *mut *mut C_tensor,
9534 out_: *mut C_tensor,
9535 x_: *mut C_tensor,
9536 y_: *mut C_tensor,
9537 dim_: i64,
9538 );
9539 pub fn atg_linear(
9540 out__: *mut *mut C_tensor,
9541 input_: *mut C_tensor,
9542 weight_: *mut C_tensor,
9543 bias_: *mut C_tensor,
9544 );
9545 pub fn atg_linear_out(
9546 out__: *mut *mut C_tensor,
9547 out_: *mut C_tensor,
9548 input_: *mut C_tensor,
9549 weight_: *mut C_tensor,
9550 bias_: *mut C_tensor,
9551 );
9552 pub fn atg_linspace(
9553 out__: *mut *mut C_tensor,
9554 start_: *mut C_scalar,
9555 end_: *mut C_scalar,
9556 steps_: i64,
9557 options_kind: c_int,
9558 options_device: c_int,
9559 );
9560 pub fn atg_linspace_out(
9561 out__: *mut *mut C_tensor,
9562 out_: *mut C_tensor,
9563 start_: *mut C_scalar,
9564 end_: *mut C_scalar,
9565 steps_: i64,
9566 );
9567 pub fn atg_linspace_scalar_tensor(
9568 out__: *mut *mut C_tensor,
9569 start_: *mut C_scalar,
9570 end_: *mut C_tensor,
9571 steps_: i64,
9572 options_kind: c_int,
9573 options_device: c_int,
9574 );
9575 pub fn atg_linspace_scalar_tensor_out(
9576 out__: *mut *mut C_tensor,
9577 out_: *mut C_tensor,
9578 start_: *mut C_scalar,
9579 end_: *mut C_tensor,
9580 steps_: i64,
9581 );
9582 pub fn atg_linspace_tensor_scalar(
9583 out__: *mut *mut C_tensor,
9584 start_: *mut C_tensor,
9585 end_: *mut C_scalar,
9586 steps_: i64,
9587 options_kind: c_int,
9588 options_device: c_int,
9589 );
9590 pub fn atg_linspace_tensor_scalar_out(
9591 out__: *mut *mut C_tensor,
9592 out_: *mut C_tensor,
9593 start_: *mut C_tensor,
9594 end_: *mut C_scalar,
9595 steps_: i64,
9596 );
9597 pub fn atg_linspace_tensor_tensor(
9598 out__: *mut *mut C_tensor,
9599 start_: *mut C_tensor,
9600 end_: *mut C_tensor,
9601 steps_: i64,
9602 options_kind: c_int,
9603 options_device: c_int,
9604 );
9605 pub fn atg_linspace_tensor_tensor_out(
9606 out__: *mut *mut C_tensor,
9607 out_: *mut C_tensor,
9608 start_: *mut C_tensor,
9609 end_: *mut C_tensor,
9610 steps_: i64,
9611 );
9612 pub fn atg_log(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9613 pub fn atg_log10(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9614 pub fn atg_log10_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9615 pub fn atg_log10_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
9616 pub fn atg_log1p(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9617 pub fn atg_log1p_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9618 pub fn atg_log1p_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
9619 pub fn atg_log2(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9620 pub fn atg_log2_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9621 pub fn atg_log2_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
9622 pub fn atg_log_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9623 pub fn atg_log_normal(out__: *mut *mut C_tensor, self_: *mut C_tensor, mean_: f64, std_: f64);
9624 pub fn atg_log_normal_(out__: *mut *mut C_tensor, self_: *mut C_tensor, mean_: f64, std_: f64);
9625 pub fn atg_log_normal_out(
9626 out__: *mut *mut C_tensor,
9627 out_: *mut C_tensor,
9628 self_: *mut C_tensor,
9629 mean_: f64,
9630 std_: f64,
9631 );
9632 pub fn atg_log_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
9633 pub fn atg_log_sigmoid(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9634 pub fn atg_log_sigmoid_backward(
9635 out__: *mut *mut C_tensor,
9636 grad_output_: *mut C_tensor,
9637 self_: *mut C_tensor,
9638 buffer_: *mut C_tensor,
9639 );
9640 pub fn atg_log_sigmoid_backward_grad_input(
9641 out__: *mut *mut C_tensor,
9642 grad_input_: *mut C_tensor,
9643 grad_output_: *mut C_tensor,
9644 self_: *mut C_tensor,
9645 buffer_: *mut C_tensor,
9646 );
9647 pub fn atg_log_sigmoid_out(
9648 out__: *mut *mut C_tensor,
9649 out_: *mut C_tensor,
9650 self_: *mut C_tensor,
9651 );
9652 pub fn atg_log_softmax(
9653 out__: *mut *mut C_tensor,
9654 self_: *mut C_tensor,
9655 dim_: i64,
9656 dtype_: c_int,
9657 );
9658 pub fn atg_log_softmax_int_out(
9659 out__: *mut *mut C_tensor,
9660 out_: *mut C_tensor,
9661 self_: *mut C_tensor,
9662 dim_: i64,
9663 dtype_: c_int,
9664 );
9665 pub fn atg_logaddexp(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9666 pub fn atg_logaddexp2(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9667 pub fn atg_logaddexp2_out(
9668 out__: *mut *mut C_tensor,
9669 out_: *mut C_tensor,
9670 self_: *mut C_tensor,
9671 other_: *mut C_tensor,
9672 );
9673 pub fn atg_logaddexp_out(
9674 out__: *mut *mut C_tensor,
9675 out_: *mut C_tensor,
9676 self_: *mut C_tensor,
9677 other_: *mut C_tensor,
9678 );
9679 pub fn atg_logcumsumexp(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
9680 pub fn atg_logcumsumexp_out(
9681 out__: *mut *mut C_tensor,
9682 out_: *mut C_tensor,
9683 self_: *mut C_tensor,
9684 dim_: i64,
9685 );
9686 pub fn atg_logdet(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9687 pub fn atg_logical_and(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9688 pub fn atg_logical_and_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9689 pub fn atg_logical_and_out(
9690 out__: *mut *mut C_tensor,
9691 out_: *mut C_tensor,
9692 self_: *mut C_tensor,
9693 other_: *mut C_tensor,
9694 );
9695 pub fn atg_logical_not(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9696 pub fn atg_logical_not_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
9697 pub fn atg_logical_not_out(
9698 out__: *mut *mut C_tensor,
9699 out_: *mut C_tensor,
9700 self_: *mut C_tensor,
9701 );
9702 pub fn atg_logical_or(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9703 pub fn atg_logical_or_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9704 pub fn atg_logical_or_out(
9705 out__: *mut *mut C_tensor,
9706 out_: *mut C_tensor,
9707 self_: *mut C_tensor,
9708 other_: *mut C_tensor,
9709 );
9710 pub fn atg_logical_xor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9711 pub fn atg_logical_xor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9712 pub fn atg_logical_xor_out(
9713 out__: *mut *mut C_tensor,
9714 out_: *mut C_tensor,
9715 self_: *mut C_tensor,
9716 other_: *mut C_tensor,
9717 );
9718 pub fn atg_logit(out__: *mut *mut C_tensor, self_: *mut C_tensor, eps_v: f64, eps_null: i8);
9719 pub fn atg_logit_(out__: *mut *mut C_tensor, self_: *mut C_tensor, eps_v: f64, eps_null: i8);
9720 pub fn atg_logit_backward(
9721 out__: *mut *mut C_tensor,
9722 grad_output_: *mut C_tensor,
9723 self_: *mut C_tensor,
9724 eps_v: f64,
9725 eps_null: i8,
9726 );
9727 pub fn atg_logit_backward_grad_input(
9728 out__: *mut *mut C_tensor,
9729 grad_input_: *mut C_tensor,
9730 grad_output_: *mut C_tensor,
9731 self_: *mut C_tensor,
9732 eps_v: f64,
9733 eps_null: i8,
9734 );
9735 pub fn atg_logit_out(
9736 out__: *mut *mut C_tensor,
9737 out_: *mut C_tensor,
9738 self_: *mut C_tensor,
9739 eps_v: f64,
9740 eps_null: i8,
9741 );
9742 pub fn atg_logspace(
9743 out__: *mut *mut C_tensor,
9744 start_: *mut C_scalar,
9745 end_: *mut C_scalar,
9746 steps_: i64,
9747 base_: f64,
9748 options_kind: c_int,
9749 options_device: c_int,
9750 );
9751 pub fn atg_logspace_out(
9752 out__: *mut *mut C_tensor,
9753 out_: *mut C_tensor,
9754 start_: *mut C_scalar,
9755 end_: *mut C_scalar,
9756 steps_: i64,
9757 base_: f64,
9758 );
9759 pub fn atg_logspace_scalar_tensor(
9760 out__: *mut *mut C_tensor,
9761 start_: *mut C_scalar,
9762 end_: *mut C_tensor,
9763 steps_: i64,
9764 base_: f64,
9765 options_kind: c_int,
9766 options_device: c_int,
9767 );
9768 pub fn atg_logspace_scalar_tensor_out(
9769 out__: *mut *mut C_tensor,
9770 out_: *mut C_tensor,
9771 start_: *mut C_scalar,
9772 end_: *mut C_tensor,
9773 steps_: i64,
9774 base_: f64,
9775 );
9776 pub fn atg_logspace_tensor_scalar(
9777 out__: *mut *mut C_tensor,
9778 start_: *mut C_tensor,
9779 end_: *mut C_scalar,
9780 steps_: i64,
9781 base_: f64,
9782 options_kind: c_int,
9783 options_device: c_int,
9784 );
9785 pub fn atg_logspace_tensor_scalar_out(
9786 out__: *mut *mut C_tensor,
9787 out_: *mut C_tensor,
9788 start_: *mut C_tensor,
9789 end_: *mut C_scalar,
9790 steps_: i64,
9791 base_: f64,
9792 );
9793 pub fn atg_logspace_tensor_tensor(
9794 out__: *mut *mut C_tensor,
9795 start_: *mut C_tensor,
9796 end_: *mut C_tensor,
9797 steps_: i64,
9798 base_: f64,
9799 options_kind: c_int,
9800 options_device: c_int,
9801 );
9802 pub fn atg_logspace_tensor_tensor_out(
9803 out__: *mut *mut C_tensor,
9804 out_: *mut C_tensor,
9805 start_: *mut C_tensor,
9806 end_: *mut C_tensor,
9807 steps_: i64,
9808 base_: f64,
9809 );
9810 pub fn atg_logsumexp(
9811 out__: *mut *mut C_tensor,
9812 self_: *mut C_tensor,
9813 dim_data: *const i64,
9814 dim_len: c_int,
9815 keepdim_: c_int,
9816 );
9817 pub fn atg_logsumexp_out(
9818 out__: *mut *mut C_tensor,
9819 out_: *mut C_tensor,
9820 self_: *mut C_tensor,
9821 dim_data: *const i64,
9822 dim_len: c_int,
9823 keepdim_: c_int,
9824 );
9825 pub fn atg_lstm(
9826 out__: *mut *mut C_tensor,
9827 input_: *mut C_tensor,
9828 hx_data: *const *mut C_tensor,
9829 hx_len: c_int,
9830 params_data: *const *mut C_tensor,
9831 params_len: c_int,
9832 has_biases_: c_int,
9833 num_layers_: i64,
9834 dropout_: f64,
9835 train_: c_int,
9836 bidirectional_: c_int,
9837 batch_first_: c_int,
9838 );
9839 pub fn atg_lstm_cell(
9840 out__: *mut *mut C_tensor,
9841 input_: *mut C_tensor,
9842 hx_data: *const *mut C_tensor,
9843 hx_len: c_int,
9844 w_ih_: *mut C_tensor,
9845 w_hh_: *mut C_tensor,
9846 b_ih_: *mut C_tensor,
9847 b_hh_: *mut C_tensor,
9848 );
9849 pub fn atg_lstm_data(
9850 out__: *mut *mut C_tensor,
9851 data_: *mut C_tensor,
9852 batch_sizes_: *mut C_tensor,
9853 hx_data: *const *mut C_tensor,
9854 hx_len: c_int,
9855 params_data: *const *mut C_tensor,
9856 params_len: c_int,
9857 has_biases_: c_int,
9858 num_layers_: i64,
9859 dropout_: f64,
9860 train_: c_int,
9861 bidirectional_: c_int,
9862 );
9863 pub fn atg_lstm_mps_backward(
9864 out0_: *mut C_tensor,
9865 out1_data: *const *mut C_tensor,
9866 out1_len: c_int,
9867 out2_data: *const *mut C_tensor,
9868 out2_len: c_int,
9869 grad_y_: *mut C_tensor,
9870 grad_hy_: *mut C_tensor,
9871 grad_cy_: *mut C_tensor,
9872 z_state_: *mut C_tensor,
9873 cell_state_fwd_: *mut C_tensor,
9874 input_: *mut C_tensor,
9875 layersOutputs_: *mut C_tensor,
9876 hx_data: *const *mut C_tensor,
9877 hx_len: c_int,
9878 params_data: *const *mut C_tensor,
9879 params_len: c_int,
9880 has_biases_: c_int,
9881 num_layers_: i64,
9882 dropout_: f64,
9883 train_: c_int,
9884 bidirectional_: c_int,
9885 batch_first_: c_int,
9886 );
9887 pub fn atg_lt(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
9888 pub fn atg_lt_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
9889 pub fn atg_lt_scalar_out(
9890 out__: *mut *mut C_tensor,
9891 out_: *mut C_tensor,
9892 self_: *mut C_tensor,
9893 other_: *mut C_scalar,
9894 );
9895 pub fn atg_lt_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9896 pub fn atg_lt_tensor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
9897 pub fn atg_lt_tensor_out(
9898 out__: *mut *mut C_tensor,
9899 out_: *mut C_tensor,
9900 self_: *mut C_tensor,
9901 other_: *mut C_tensor,
9902 );
9903 pub fn atg_lu_solve(
9904 out__: *mut *mut C_tensor,
9905 self_: *mut C_tensor,
9906 LU_data_: *mut C_tensor,
9907 LU_pivots_: *mut C_tensor,
9908 );
9909 pub fn atg_lu_solve_out(
9910 out__: *mut *mut C_tensor,
9911 out_: *mut C_tensor,
9912 self_: *mut C_tensor,
9913 LU_data_: *mut C_tensor,
9914 LU_pivots_: *mut C_tensor,
9915 );
9916 pub fn atg_lu_unpack(
9917 out__: *mut *mut C_tensor,
9918 LU_data_: *mut C_tensor,
9919 LU_pivots_: *mut C_tensor,
9920 unpack_data_: c_int,
9921 unpack_pivots_: c_int,
9922 );
9923 pub fn atg_lu_unpack_out(
9924 out__: *mut *mut C_tensor,
9925 P_: *mut C_tensor,
9926 L_: *mut C_tensor,
9927 U_: *mut C_tensor,
9928 LU_data_: *mut C_tensor,
9929 LU_pivots_: *mut C_tensor,
9930 unpack_data_: c_int,
9931 unpack_pivots_: c_int,
9932 );
9933 pub fn atg_margin_ranking_loss(
9934 out__: *mut *mut C_tensor,
9935 input1_: *mut C_tensor,
9936 input2_: *mut C_tensor,
9937 target_: *mut C_tensor,
9938 margin_: f64,
9939 reduction_: i64,
9940 );
9941 pub fn atg_masked_fill(
9942 out__: *mut *mut C_tensor,
9943 self_: *mut C_tensor,
9944 mask_: *mut C_tensor,
9945 value_: *mut C_scalar,
9946 );
9947 pub fn atg_masked_fill_(
9948 out__: *mut *mut C_tensor,
9949 self_: *mut C_tensor,
9950 mask_: *mut C_tensor,
9951 value_: *mut C_scalar,
9952 );
9953 pub fn atg_masked_fill_scalar_out(
9954 out__: *mut *mut C_tensor,
9955 out_: *mut C_tensor,
9956 self_: *mut C_tensor,
9957 mask_: *mut C_tensor,
9958 value_: *mut C_scalar,
9959 );
9960 pub fn atg_masked_fill_tensor(
9961 out__: *mut *mut C_tensor,
9962 self_: *mut C_tensor,
9963 mask_: *mut C_tensor,
9964 value_: *mut C_tensor,
9965 );
9966 pub fn atg_masked_fill_tensor_(
9967 out__: *mut *mut C_tensor,
9968 self_: *mut C_tensor,
9969 mask_: *mut C_tensor,
9970 value_: *mut C_tensor,
9971 );
9972 pub fn atg_masked_fill_tensor_out(
9973 out__: *mut *mut C_tensor,
9974 out_: *mut C_tensor,
9975 self_: *mut C_tensor,
9976 mask_: *mut C_tensor,
9977 value_: *mut C_tensor,
9978 );
9979 pub fn atg_masked_scatter(
9980 out__: *mut *mut C_tensor,
9981 self_: *mut C_tensor,
9982 mask_: *mut C_tensor,
9983 source_: *mut C_tensor,
9984 );
9985 pub fn atg_masked_scatter_(
9986 out__: *mut *mut C_tensor,
9987 self_: *mut C_tensor,
9988 mask_: *mut C_tensor,
9989 source_: *mut C_tensor,
9990 );
9991 pub fn atg_masked_scatter_backward(
9992 out__: *mut *mut C_tensor,
9993 grad_output_: *mut C_tensor,
9994 mask_: *mut C_tensor,
9995 sizes_data: *const i64,
9996 sizes_len: c_int,
9997 );
9998 pub fn atg_masked_scatter_out(
9999 out__: *mut *mut C_tensor,
10000 out_: *mut C_tensor,
10001 self_: *mut C_tensor,
10002 mask_: *mut C_tensor,
10003 source_: *mut C_tensor,
10004 );
10005 pub fn atg_masked_select(out__: *mut *mut C_tensor, self_: *mut C_tensor, mask_: *mut C_tensor);
10006 pub fn atg_masked_select_backward(
10007 out__: *mut *mut C_tensor,
10008 grad_: *mut C_tensor,
10009 input_: *mut C_tensor,
10010 mask_: *mut C_tensor,
10011 );
10012 pub fn atg_masked_select_out(
10013 out__: *mut *mut C_tensor,
10014 out_: *mut C_tensor,
10015 self_: *mut C_tensor,
10016 mask_: *mut C_tensor,
10017 );
10018 pub fn atg_matmul(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
10019 pub fn atg_matmul_out(
10020 out__: *mut *mut C_tensor,
10021 out_: *mut C_tensor,
10022 self_: *mut C_tensor,
10023 other_: *mut C_tensor,
10024 );
10025 pub fn atg_matrix_exp(out__: *mut *mut C_tensor, self_: *mut C_tensor);
10026 pub fn atg_matrix_exp_backward(
10027 out__: *mut *mut C_tensor,
10028 self_: *mut C_tensor,
10029 grad_: *mut C_tensor,
10030 );
10031 pub fn atg_matrix_h(out__: *mut *mut C_tensor, self_: *mut C_tensor);
10032 pub fn atg_matrix_power(out__: *mut *mut C_tensor, self_: *mut C_tensor, n_: i64);
10033 pub fn atg_matrix_power_out(
10034 out__: *mut *mut C_tensor,
10035 out_: *mut C_tensor,
10036 self_: *mut C_tensor,
10037 n_: i64,
10038 );
10039 pub fn atg_max(out__: *mut *mut C_tensor, self_: *mut C_tensor);
10040 pub fn atg_max_dim(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, keepdim_: c_int);
10041 pub fn atg_max_dim_max(
10042 out__: *mut *mut C_tensor,
10043 max_: *mut C_tensor,
10044 max_values_: *mut C_tensor,
10045 self_: *mut C_tensor,
10046 dim_: i64,
10047 keepdim_: c_int,
10048 );
10049 pub fn atg_max_other(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
10050 pub fn atg_max_out(
10051 out__: *mut *mut C_tensor,
10052 out_: *mut C_tensor,
10053 self_: *mut C_tensor,
10054 other_: *mut C_tensor,
10055 );
10056 pub fn atg_max_pool1d(
10057 out__: *mut *mut C_tensor,
10058 self_: *mut C_tensor,
10059 kernel_size_data: *const i64,
10060 kernel_size_len: c_int,
10061 stride_data: *const i64,
10062 stride_len: c_int,
10063 padding_data: *const i64,
10064 padding_len: c_int,
10065 dilation_data: *const i64,
10066 dilation_len: c_int,
10067 ceil_mode_: c_int,
10068 );
10069 pub fn atg_max_pool1d_with_indices(
10070 out__: *mut *mut C_tensor,
10071 self_: *mut C_tensor,
10072 kernel_size_data: *const i64,
10073 kernel_size_len: c_int,
10074 stride_data: *const i64,
10075 stride_len: c_int,
10076 padding_data: *const i64,
10077 padding_len: c_int,
10078 dilation_data: *const i64,
10079 dilation_len: c_int,
10080 ceil_mode_: c_int,
10081 );
10082 pub fn atg_max_pool2d(
10083 out__: *mut *mut C_tensor,
10084 self_: *mut C_tensor,
10085 kernel_size_data: *const i64,
10086 kernel_size_len: c_int,
10087 stride_data: *const i64,
10088 stride_len: c_int,
10089 padding_data: *const i64,
10090 padding_len: c_int,
10091 dilation_data: *const i64,
10092 dilation_len: c_int,
10093 ceil_mode_: c_int,
10094 );
10095 pub fn atg_max_pool2d_backward(
10096 out__: *mut *mut C_tensor,
10097 grad_output_: *mut C_tensor,
10098 self_: *mut C_tensor,
10099 kernel_size_data: *const i64,
10100 kernel_size_len: c_int,
10101 stride_data: *const i64,
10102 stride_len: c_int,
10103 padding_data: *const i64,
10104 padding_len: c_int,
10105 dilation_data: *const i64,
10106 dilation_len: c_int,
10107 ceil_mode_: c_int,
10108 );
10109 pub fn atg_max_pool2d_backward_out(
10110 out__: *mut *mut C_tensor,
10111 out_: *mut C_tensor,
10112 grad_output_: *mut C_tensor,
10113 self_: *mut C_tensor,
10114 kernel_size_data: *const i64,
10115 kernel_size_len: c_int,
10116 stride_data: *const i64,
10117 stride_len: c_int,
10118 padding_data: *const i64,
10119 padding_len: c_int,
10120 dilation_data: *const i64,
10121 dilation_len: c_int,
10122 ceil_mode_: c_int,
10123 );
10124 pub fn atg_max_pool2d_with_indices(
10125 out__: *mut *mut C_tensor,
10126 self_: *mut C_tensor,
10127 kernel_size_data: *const i64,
10128 kernel_size_len: c_int,
10129 stride_data: *const i64,
10130 stride_len: c_int,
10131 padding_data: *const i64,
10132 padding_len: c_int,
10133 dilation_data: *const i64,
10134 dilation_len: c_int,
10135 ceil_mode_: c_int,
10136 );
10137 pub fn atg_max_pool2d_with_indices_backward(
10138 out__: *mut *mut C_tensor,
10139 grad_output_: *mut C_tensor,
10140 self_: *mut C_tensor,
10141 kernel_size_data: *const i64,
10142 kernel_size_len: c_int,
10143 stride_data: *const i64,
10144 stride_len: c_int,
10145 padding_data: *const i64,
10146 padding_len: c_int,
10147 dilation_data: *const i64,
10148 dilation_len: c_int,
10149 ceil_mode_: c_int,
10150 indices_: *mut C_tensor,
10151 );
10152 pub fn atg_max_pool2d_with_indices_backward_grad_input(
10153 out__: *mut *mut C_tensor,
10154 grad_input_: *mut C_tensor,
10155 grad_output_: *mut C_tensor,
10156 self_: *mut C_tensor,
10157 kernel_size_data: *const i64,
10158 kernel_size_len: c_int,
10159 stride_data: *const i64,
10160 stride_len: c_int,
10161 padding_data: *const i64,
10162 padding_len: c_int,
10163 dilation_data: *const i64,
10164 dilation_len: c_int,
10165 ceil_mode_: c_int,
10166 indices_: *mut C_tensor,
10167 );
10168 pub fn atg_max_pool2d_with_indices_out(
10169 out__: *mut *mut C_tensor,
10170 out_: *mut C_tensor,
10171 indices_: *mut C_tensor,
10172 self_: *mut C_tensor,
10173 kernel_size_data: *const i64,
10174 kernel_size_len: c_int,
10175 stride_data: *const i64,
10176 stride_len: c_int,
10177 padding_data: *const i64,
10178 padding_len: c_int,
10179 dilation_data: *const i64,
10180 dilation_len: c_int,
10181 ceil_mode_: c_int,
10182 );
10183 pub fn atg_max_pool3d(
10184 out__: *mut *mut C_tensor,
10185 self_: *mut C_tensor,
10186 kernel_size_data: *const i64,
10187 kernel_size_len: c_int,
10188 stride_data: *const i64,
10189 stride_len: c_int,
10190 padding_data: *const i64,
10191 padding_len: c_int,
10192 dilation_data: *const i64,
10193 dilation_len: c_int,
10194 ceil_mode_: c_int,
10195 );
10196 pub fn atg_max_pool3d_with_indices(
10197 out__: *mut *mut C_tensor,
10198 self_: *mut C_tensor,
10199 kernel_size_data: *const i64,
10200 kernel_size_len: c_int,
10201 stride_data: *const i64,
10202 stride_len: c_int,
10203 padding_data: *const i64,
10204 padding_len: c_int,
10205 dilation_data: *const i64,
10206 dilation_len: c_int,
10207 ceil_mode_: c_int,
10208 );
10209 pub fn atg_max_pool3d_with_indices_backward(
10210 out__: *mut *mut C_tensor,
10211 grad_output_: *mut C_tensor,
10212 self_: *mut C_tensor,
10213 kernel_size_data: *const i64,
10214 kernel_size_len: c_int,
10215 stride_data: *const i64,
10216 stride_len: c_int,
10217 padding_data: *const i64,
10218 padding_len: c_int,
10219 dilation_data: *const i64,
10220 dilation_len: c_int,
10221 ceil_mode_: c_int,
10222 indices_: *mut C_tensor,
10223 );
10224 pub fn atg_max_pool3d_with_indices_backward_grad_input(
10225 out__: *mut *mut C_tensor,
10226 grad_input_: *mut C_tensor,
10227 grad_output_: *mut C_tensor,
10228 self_: *mut C_tensor,
10229 kernel_size_data: *const i64,
10230 kernel_size_len: c_int,
10231 stride_data: *const i64,
10232 stride_len: c_int,
10233 padding_data: *const i64,
10234 padding_len: c_int,
10235 dilation_data: *const i64,
10236 dilation_len: c_int,
10237 ceil_mode_: c_int,
10238 indices_: *mut C_tensor,
10239 );
10240 pub fn atg_max_pool3d_with_indices_out(
10241 out__: *mut *mut C_tensor,
10242 out_: *mut C_tensor,
10243 indices_: *mut C_tensor,
10244 self_: *mut C_tensor,
10245 kernel_size_data: *const i64,
10246 kernel_size_len: c_int,
10247 stride_data: *const i64,
10248 stride_len: c_int,
10249 padding_data: *const i64,
10250 padding_len: c_int,
10251 dilation_data: *const i64,
10252 dilation_len: c_int,
10253 ceil_mode_: c_int,
10254 );
10255 pub fn atg_max_unary_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
10256 pub fn atg_max_unpool2d(
10257 out__: *mut *mut C_tensor,
10258 self_: *mut C_tensor,
10259 indices_: *mut C_tensor,
10260 output_size_data: *const i64,
10261 output_size_len: c_int,
10262 );
10263 pub fn atg_max_unpool2d_out(
10264 out__: *mut *mut C_tensor,
10265 out_: *mut C_tensor,
10266 self_: *mut C_tensor,
10267 indices_: *mut C_tensor,
10268 output_size_data: *const i64,
10269 output_size_len: c_int,
10270 );
10271 pub fn atg_max_unpool3d(
10272 out__: *mut *mut C_tensor,
10273 self_: *mut C_tensor,
10274 indices_: *mut C_tensor,
10275 output_size_data: *const i64,
10276 output_size_len: c_int,
10277 stride_data: *const i64,
10278 stride_len: c_int,
10279 padding_data: *const i64,
10280 padding_len: c_int,
10281 );
10282 pub fn atg_max_unpool3d_out(
10283 out__: *mut *mut C_tensor,
10284 out_: *mut C_tensor,
10285 self_: *mut C_tensor,
10286 indices_: *mut C_tensor,
10287 output_size_data: *const i64,
10288 output_size_len: c_int,
10289 stride_data: *const i64,
10290 stride_len: c_int,
10291 padding_data: *const i64,
10292 padding_len: c_int,
10293 );
10294 pub fn atg_maximum(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
10295 pub fn atg_maximum_out(
10296 out__: *mut *mut C_tensor,
10297 out_: *mut C_tensor,
10298 self_: *mut C_tensor,
10299 other_: *mut C_tensor,
10300 );
10301 pub fn atg_mean(out__: *mut *mut C_tensor, self_: *mut C_tensor, dtype_: c_int);
10302 pub fn atg_mean_dim(
10303 out__: *mut *mut C_tensor,
10304 self_: *mut C_tensor,
10305 dim_data: *const i64,
10306 dim_len: c_int,
10307 keepdim_: c_int,
10308 dtype_: c_int,
10309 );
10310 pub fn atg_mean_dtype_out(
10311 out__: *mut *mut C_tensor,
10312 out_: *mut C_tensor,
10313 self_: *mut C_tensor,
10314 dtype_: c_int,
10315 );
10316 pub fn atg_mean_out(
10317 out__: *mut *mut C_tensor,
10318 out_: *mut C_tensor,
10319 self_: *mut C_tensor,
10320 dim_data: *const i64,
10321 dim_len: c_int,
10322 keepdim_: c_int,
10323 dtype_: c_int,
10324 );
10325 pub fn atg_median(out__: *mut *mut C_tensor, self_: *mut C_tensor);
10326 pub fn atg_median_dim(
10327 out__: *mut *mut C_tensor,
10328 self_: *mut C_tensor,
10329 dim_: i64,
10330 keepdim_: c_int,
10331 );
10332 pub fn atg_median_dim_values(
10333 out__: *mut *mut C_tensor,
10334 values_: *mut C_tensor,
10335 indices_: *mut C_tensor,
10336 self_: *mut C_tensor,
10337 dim_: i64,
10338 keepdim_: c_int,
10339 );
10340 pub fn atg_median_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
10341 pub fn atg_meshgrid(
10342 tensors_data: *const *mut C_tensor,
10343 tensors_len: c_int,
10344 ) -> *mut *mut C_tensor;
10345 pub fn atg_meshgrid_indexing(
10346 tensors_data: *const *mut C_tensor,
10347 tensors_len: c_int,
10348 indexing_ptr: *const u8,
10349 indexing_len: c_int,
10350 ) -> *mut *mut C_tensor;
10351 pub fn atg_mh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
10352 pub fn atg_min(out__: *mut *mut C_tensor, self_: *mut C_tensor);
10353 pub fn atg_min_dim(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, keepdim_: c_int);
10354 pub fn atg_min_dim_min(
10355 out__: *mut *mut C_tensor,
10356 min_: *mut C_tensor,
10357 min_indices_: *mut C_tensor,
10358 self_: *mut C_tensor,
10359 dim_: i64,
10360 keepdim_: c_int,
10361 );
10362 pub fn atg_min_other(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
10363 pub fn atg_min_out(
10364 out__: *mut *mut C_tensor,
10365 out_: *mut C_tensor,
10366 self_: *mut C_tensor,
10367 other_: *mut C_tensor,
10368 );
10369 pub fn atg_min_unary_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
10370 pub fn atg_minimum(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
10371 pub fn atg_minimum_out(
10372 out__: *mut *mut C_tensor,
10373 out_: *mut C_tensor,
10374 self_: *mut C_tensor,
10375 other_: *mut C_tensor,
10376 );
10377 pub fn atg_miopen_batch_norm(
10378 out__: *mut *mut C_tensor,
10379 input_: *mut C_tensor,
10380 weight_: *mut C_tensor,
10381 bias_: *mut C_tensor,
10382 running_mean_: *mut C_tensor,
10383 running_var_: *mut C_tensor,
10384 training_: c_int,
10385 exponential_average_factor_: f64,
10386 epsilon_: f64,
10387 );
10388 pub fn atg_miopen_batch_norm_backward(
10389 out__: *mut *mut C_tensor,
10390 input_: *mut C_tensor,
10391 grad_output_: *mut C_tensor,
10392 weight_: *mut C_tensor,
10393 running_mean_: *mut C_tensor,
10394 running_var_: *mut C_tensor,
10395 save_mean_: *mut C_tensor,
10396 save_var_: *mut C_tensor,
10397 epsilon_: f64,
10398 );
10399 pub fn atg_miopen_batch_norm_backward_out(
10400 out__: *mut *mut C_tensor,
10401 out0_: *mut C_tensor,
10402 out1_: *mut C_tensor,
10403 out2_: *mut C_tensor,
10404 input_: *mut C_tensor,
10405 grad_output_: *mut C_tensor,
10406 weight_: *mut C_tensor,
10407 running_mean_: *mut C_tensor,
10408 running_var_: *mut C_tensor,
10409 save_mean_: *mut C_tensor,
10410 save_var_: *mut C_tensor,
10411 epsilon_: f64,
10412 );
10413 pub fn atg_miopen_batch_norm_out(
10414 out__: *mut *mut C_tensor,
10415 out0_: *mut C_tensor,
10416 out1_: *mut C_tensor,
10417 out2_: *mut C_tensor,
10418 input_: *mut C_tensor,
10419 weight_: *mut C_tensor,
10420 bias_: *mut C_tensor,
10421 running_mean_: *mut C_tensor,
10422 running_var_: *mut C_tensor,
10423 training_: c_int,
10424 exponential_average_factor_: f64,
10425 epsilon_: f64,
10426 );
10427 pub fn atg_miopen_convolution(
10428 out__: *mut *mut C_tensor,
10429 self_: *mut C_tensor,
10430 weight_: *mut C_tensor,
10431 bias_: *mut C_tensor,
10432 padding_data: *const i64,
10433 padding_len: c_int,
10434 stride_data: *const i64,
10435 stride_len: c_int,
10436 dilation_data: *const i64,
10437 dilation_len: c_int,
10438 groups_: i64,
10439 benchmark_: c_int,
10440 deterministic_: c_int,
10441 );
10442 pub fn atg_miopen_convolution_add_relu(
10443 out__: *mut *mut C_tensor,
10444 self_: *mut C_tensor,
10445 weight_: *mut C_tensor,
10446 z_: *mut C_tensor,
10447 alpha_: *mut C_scalar,
10448 bias_: *mut C_tensor,
10449 stride_data: *const i64,
10450 stride_len: c_int,
10451 padding_data: *const i64,
10452 padding_len: c_int,
10453 dilation_data: *const i64,
10454 dilation_len: c_int,
10455 groups_: i64,
10456 );
10457 pub fn atg_miopen_convolution_out(
10458 out__: *mut *mut C_tensor,
10459 out_: *mut C_tensor,
10460 self_: *mut C_tensor,
10461 weight_: *mut C_tensor,
10462 bias_: *mut C_tensor,
10463 padding_data: *const i64,
10464 padding_len: c_int,
10465 stride_data: *const i64,
10466 stride_len: c_int,
10467 dilation_data: *const i64,
10468 dilation_len: c_int,
10469 groups_: i64,
10470 benchmark_: c_int,
10471 deterministic_: c_int,
10472 );
10473 pub fn atg_miopen_convolution_relu(
10474 out__: *mut *mut C_tensor,
10475 self_: *mut C_tensor,
10476 weight_: *mut C_tensor,
10477 bias_: *mut C_tensor,
10478 stride_data: *const i64,
10479 stride_len: c_int,
10480 padding_data: *const i64,
10481 padding_len: c_int,
10482 dilation_data: *const i64,
10483 dilation_len: c_int,
10484 groups_: i64,
10485 );
10486 pub fn atg_miopen_convolution_transpose(
10487 out__: *mut *mut C_tensor,
10488 self_: *mut C_tensor,
10489 weight_: *mut C_tensor,
10490 bias_: *mut C_tensor,
10491 padding_data: *const i64,
10492 padding_len: c_int,
10493 output_padding_data: *const i64,
10494 output_padding_len: c_int,
10495 stride_data: *const i64,
10496 stride_len: c_int,
10497 dilation_data: *const i64,
10498 dilation_len: c_int,
10499 groups_: i64,
10500 benchmark_: c_int,
10501 deterministic_: c_int,
10502 );
10503 pub fn atg_miopen_convolution_transpose_out(
10504 out__: *mut *mut C_tensor,
10505 out_: *mut C_tensor,
10506 self_: *mut C_tensor,
10507 weight_: *mut C_tensor,
10508 bias_: *mut C_tensor,
10509 padding_data: *const i64,
10510 padding_len: c_int,
10511 output_padding_data: *const i64,
10512 output_padding_len: c_int,
10513 stride_data: *const i64,
10514 stride_len: c_int,
10515 dilation_data: *const i64,
10516 dilation_len: c_int,
10517 groups_: i64,
10518 benchmark_: c_int,
10519 deterministic_: c_int,
10520 );
10521 pub fn atg_miopen_ctc_loss(
10522 out__: *mut *mut C_tensor,
10523 log_probs_: *mut C_tensor,
10524 targets_: *mut C_tensor,
10525 input_lengths_data: *const i64,
10526 input_lengths_len: c_int,
10527 target_lengths_data: *const i64,
10528 target_lengths_len: c_int,
10529 blank_: i64,
10530 deterministic_: c_int,
10531 zero_infinity_: c_int,
10532 );
10533 pub fn atg_miopen_ctc_loss_out(
10534 out__: *mut *mut C_tensor,
10535 out0_: *mut C_tensor,
10536 out1_: *mut C_tensor,
10537 log_probs_: *mut C_tensor,
10538 targets_: *mut C_tensor,
10539 input_lengths_data: *const i64,
10540 input_lengths_len: c_int,
10541 target_lengths_data: *const i64,
10542 target_lengths_len: c_int,
10543 blank_: i64,
10544 deterministic_: c_int,
10545 zero_infinity_: c_int,
10546 );
10547 pub fn atg_miopen_ctc_loss_tensor(
10548 out__: *mut *mut C_tensor,
10549 log_probs_: *mut C_tensor,
10550 targets_: *mut C_tensor,
10551 input_lengths_: *mut C_tensor,
10552 target_lengths_: *mut C_tensor,
10553 blank_: i64,
10554 deterministic_: c_int,
10555 zero_infinity_: c_int,
10556 );
10557 pub fn atg_miopen_depthwise_convolution(
10558 out__: *mut *mut C_tensor,
10559 self_: *mut C_tensor,
10560 weight_: *mut C_tensor,
10561 bias_: *mut C_tensor,
10562 padding_data: *const i64,
10563 padding_len: c_int,
10564 stride_data: *const i64,
10565 stride_len: c_int,
10566 dilation_data: *const i64,
10567 dilation_len: c_int,
10568 groups_: i64,
10569 benchmark_: c_int,
10570 deterministic_: c_int,
10571 );
10572 pub fn atg_miopen_depthwise_convolution_out(
10573 out__: *mut *mut C_tensor,
10574 out_: *mut C_tensor,
10575 self_: *mut C_tensor,
10576 weight_: *mut C_tensor,
10577 bias_: *mut C_tensor,
10578 padding_data: *const i64,
10579 padding_len: c_int,
10580 stride_data: *const i64,
10581 stride_len: c_int,
10582 dilation_data: *const i64,
10583 dilation_len: c_int,
10584 groups_: i64,
10585 benchmark_: c_int,
10586 deterministic_: c_int,
10587 );
10588 pub fn atg_miopen_rnn(
10589 out__: *mut *mut C_tensor,
10590 input_: *mut C_tensor,
10591 weight_data: *const *mut C_tensor,
10592 weight_len: c_int,
10593 weight_stride0_: i64,
10594 hx_: *mut C_tensor,
10595 cx_: *mut C_tensor,
10596 mode_: i64,
10597 hidden_size_: i64,
10598 num_layers_: i64,
10599 batch_first_: c_int,
10600 dropout_: f64,
10601 train_: c_int,
10602 bidirectional_: c_int,
10603 batch_sizes_data: *const i64,
10604 batch_sizes_len: c_int,
10605 dropout_state_: *mut C_tensor,
10606 );
10607 pub fn atg_miopen_rnn_out(
10608 out__: *mut *mut C_tensor,
10609 out0_: *mut C_tensor,
10610 out1_: *mut C_tensor,
10611 out2_: *mut C_tensor,
10612 out3_: *mut C_tensor,
10613 out4_: *mut C_tensor,
10614 input_: *mut C_tensor,
10615 weight_data: *const *mut C_tensor,
10616 weight_len: c_int,
10617 weight_stride0_: i64,
10618 hx_: *mut C_tensor,
10619 cx_: *mut C_tensor,
10620 mode_: i64,
10621 hidden_size_: i64,
10622 num_layers_: i64,
10623 batch_first_: c_int,
10624 dropout_: f64,
10625 train_: c_int,
10626 bidirectional_: c_int,
10627 batch_sizes_data: *const i64,
10628 batch_sizes_len: c_int,
10629 dropout_state_: *mut C_tensor,
10630 );
10631 pub fn atg_mish(out__: *mut *mut C_tensor, self_: *mut C_tensor);
10632 pub fn atg_mish_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
10633 pub fn atg_mish_backward(
10634 out__: *mut *mut C_tensor,
10635 grad_output_: *mut C_tensor,
10636 self_: *mut C_tensor,
10637 );
10638 pub fn atg_mish_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
10639 pub fn atg_mkldnn_adaptive_avg_pool2d(
10640 out__: *mut *mut C_tensor,
10641 self_: *mut C_tensor,
10642 output_size_data: *const i64,
10643 output_size_len: c_int,
10644 );
10645 pub fn atg_mkldnn_adaptive_avg_pool2d_backward(
10646 out__: *mut *mut C_tensor,
10647 grad_output_: *mut C_tensor,
10648 self_: *mut C_tensor,
10649 );
10650 pub fn atg_mkldnn_adaptive_avg_pool2d_backward_out(
10651 out__: *mut *mut C_tensor,
10652 out_: *mut C_tensor,
10653 grad_output_: *mut C_tensor,
10654 self_: *mut C_tensor,
10655 );
10656 pub fn atg_mkldnn_adaptive_avg_pool2d_out(
10657 out__: *mut *mut C_tensor,
10658 out_: *mut C_tensor,
10659 self_: *mut C_tensor,
10660 output_size_data: *const i64,
10661 output_size_len: c_int,
10662 );
10663 pub fn atg_mkldnn_convolution(
10664 out__: *mut *mut C_tensor,
10665 self_: *mut C_tensor,
10666 weight_: *mut C_tensor,
10667 bias_: *mut C_tensor,
10668 padding_data: *const i64,
10669 padding_len: c_int,
10670 stride_data: *const i64,
10671 stride_len: c_int,
10672 dilation_data: *const i64,
10673 dilation_len: c_int,
10674 groups_: i64,
10675 );
10676 pub fn atg_mkldnn_convolution_out(
10677 out__: *mut *mut C_tensor,
10678 out_: *mut C_tensor,
10679 self_: *mut C_tensor,
10680 weight_: *mut C_tensor,
10681 bias_: *mut C_tensor,
10682 padding_data: *const i64,
10683 padding_len: c_int,
10684 stride_data: *const i64,
10685 stride_len: c_int,
10686 dilation_data: *const i64,
10687 dilation_len: c_int,
10688 groups_: i64,
10689 );
10690 pub fn atg_mkldnn_linear(
10691 out__: *mut *mut C_tensor,
10692 self_: *mut C_tensor,
10693 weight_: *mut C_tensor,
10694 bias_: *mut C_tensor,
10695 );
10696 pub fn atg_mkldnn_linear_backward_input(
10697 out__: *mut *mut C_tensor,
10698 input_size_data: *const i64,
10699 input_size_len: c_int,
10700 grad_output_: *mut C_tensor,
10701 weight_: *mut C_tensor,
10702 );
10703 pub fn atg_mkldnn_linear_backward_input_out(
10704 out__: *mut *mut C_tensor,
10705 out_: *mut C_tensor,
10706 input_size_data: *const i64,
10707 input_size_len: c_int,
10708 grad_output_: *mut C_tensor,
10709 weight_: *mut C_tensor,
10710 );
10711 pub fn atg_mkldnn_linear_backward_weights(
10712 out__: *mut *mut C_tensor,
10713 grad_output_: *mut C_tensor,
10714 input_: *mut C_tensor,
10715 weight_: *mut C_tensor,
10716 bias_defined_: c_int,
10717 );
10718 pub fn atg_mkldnn_linear_backward_weights_out(
10719 out__: *mut *mut C_tensor,
10720 out0_: *mut C_tensor,
10721 out1_: *mut C_tensor,
10722 grad_output_: *mut C_tensor,
10723 input_: *mut C_tensor,
10724 weight_: *mut C_tensor,
10725 bias_defined_: c_int,
10726 );
10727 pub fn atg_mkldnn_linear_out(
10728 out__: *mut *mut C_tensor,
10729 out_: *mut C_tensor,
10730 self_: *mut C_tensor,
10731 weight_: *mut C_tensor,
10732 bias_: *mut C_tensor,
10733 );
10734 pub fn atg_mkldnn_max_pool2d(
10735 out__: *mut *mut C_tensor,
10736 self_: *mut C_tensor,
10737 kernel_size_data: *const i64,
10738 kernel_size_len: c_int,
10739 stride_data: *const i64,
10740 stride_len: c_int,
10741 padding_data: *const i64,
10742 padding_len: c_int,
10743 dilation_data: *const i64,
10744 dilation_len: c_int,
10745 ceil_mode_: c_int,
10746 );
10747 pub fn atg_mkldnn_max_pool2d_backward(
10748 out__: *mut *mut C_tensor,
10749 grad_output_: *mut C_tensor,
10750 output_: *mut C_tensor,
10751 input_: *mut C_tensor,
10752 kernel_size_data: *const i64,
10753 kernel_size_len: c_int,
10754 stride_data: *const i64,
10755 stride_len: c_int,
10756 padding_data: *const i64,
10757 padding_len: c_int,
10758 dilation_data: *const i64,
10759 dilation_len: c_int,
10760 ceil_mode_: c_int,
10761 );
10762 pub fn atg_mkldnn_max_pool2d_backward_out(
10763 out__: *mut *mut C_tensor,
10764 out_: *mut C_tensor,
10765 grad_output_: *mut C_tensor,
10766 output_: *mut C_tensor,
10767 input_: *mut C_tensor,
10768 kernel_size_data: *const i64,
10769 kernel_size_len: c_int,
10770 stride_data: *const i64,
10771 stride_len: c_int,
10772 padding_data: *const i64,
10773 padding_len: c_int,
10774 dilation_data: *const i64,
10775 dilation_len: c_int,
10776 ceil_mode_: c_int,
10777 );
10778 pub fn atg_mkldnn_max_pool2d_out(
10779 out__: *mut *mut C_tensor,
10780 out_: *mut C_tensor,
10781 self_: *mut C_tensor,
10782 kernel_size_data: *const i64,
10783 kernel_size_len: c_int,
10784 stride_data: *const i64,
10785 stride_len: c_int,
10786 padding_data: *const i64,
10787 padding_len: c_int,
10788 dilation_data: *const i64,
10789 dilation_len: c_int,
10790 ceil_mode_: c_int,
10791 );
10792 pub fn atg_mkldnn_max_pool3d(
10793 out__: *mut *mut C_tensor,
10794 self_: *mut C_tensor,
10795 kernel_size_data: *const i64,
10796 kernel_size_len: c_int,
10797 stride_data: *const i64,
10798 stride_len: c_int,
10799 padding_data: *const i64,
10800 padding_len: c_int,
10801 dilation_data: *const i64,
10802 dilation_len: c_int,
10803 ceil_mode_: c_int,
10804 );
10805 pub fn atg_mkldnn_max_pool3d_backward(
10806 out__: *mut *mut C_tensor,
10807 grad_output_: *mut C_tensor,
10808 output_: *mut C_tensor,
10809 input_: *mut C_tensor,
10810 kernel_size_data: *const i64,
10811 kernel_size_len: c_int,
10812 stride_data: *const i64,
10813 stride_len: c_int,
10814 padding_data: *const i64,
10815 padding_len: c_int,
10816 dilation_data: *const i64,
10817 dilation_len: c_int,
10818 ceil_mode_: c_int,
10819 );
10820 pub fn atg_mkldnn_max_pool3d_backward_out(
10821 out__: *mut *mut C_tensor,
10822 out_: *mut C_tensor,
10823 grad_output_: *mut C_tensor,
10824 output_: *mut C_tensor,
10825 input_: *mut C_tensor,
10826 kernel_size_data: *const i64,
10827 kernel_size_len: c_int,
10828 stride_data: *const i64,
10829 stride_len: c_int,
10830 padding_data: *const i64,
10831 padding_len: c_int,
10832 dilation_data: *const i64,
10833 dilation_len: c_int,
10834 ceil_mode_: c_int,
10835 );
10836 pub fn atg_mkldnn_max_pool3d_out(
10837 out__: *mut *mut C_tensor,
10838 out_: *mut C_tensor,
10839 self_: *mut C_tensor,
10840 kernel_size_data: *const i64,
10841 kernel_size_len: c_int,
10842 stride_data: *const i64,
10843 stride_len: c_int,
10844 padding_data: *const i64,
10845 padding_len: c_int,
10846 dilation_data: *const i64,
10847 dilation_len: c_int,
10848 ceil_mode_: c_int,
10849 );
10850 pub fn atg_mkldnn_reorder_conv2d_weight(
10851 out__: *mut *mut C_tensor,
10852 self_: *mut C_tensor,
10853 padding_data: *const i64,
10854 padding_len: c_int,
10855 stride_data: *const i64,
10856 stride_len: c_int,
10857 dilation_data: *const i64,
10858 dilation_len: c_int,
10859 groups_: i64,
10860 input_size_data: *const i64,
10861 input_size_len: c_int,
10862 );
10863 pub fn atg_mkldnn_reorder_conv2d_weight_out(
10864 out__: *mut *mut C_tensor,
10865 out_: *mut C_tensor,
10866 self_: *mut C_tensor,
10867 padding_data: *const i64,
10868 padding_len: c_int,
10869 stride_data: *const i64,
10870 stride_len: c_int,
10871 dilation_data: *const i64,
10872 dilation_len: c_int,
10873 groups_: i64,
10874 input_size_data: *const i64,
10875 input_size_len: c_int,
10876 );
10877 pub fn atg_mkldnn_reorder_conv3d_weight(
10878 out__: *mut *mut C_tensor,
10879 self_: *mut C_tensor,
10880 padding_data: *const i64,
10881 padding_len: c_int,
10882 stride_data: *const i64,
10883 stride_len: c_int,
10884 dilation_data: *const i64,
10885 dilation_len: c_int,
10886 groups_: i64,
10887 input_size_data: *const i64,
10888 input_size_len: c_int,
10889 );
10890 pub fn atg_mkldnn_reorder_conv3d_weight_out(
10891 out__: *mut *mut C_tensor,
10892 out_: *mut C_tensor,
10893 self_: *mut C_tensor,
10894 padding_data: *const i64,
10895 padding_len: c_int,
10896 stride_data: *const i64,
10897 stride_len: c_int,
10898 dilation_data: *const i64,
10899 dilation_len: c_int,
10900 groups_: i64,
10901 input_size_data: *const i64,
10902 input_size_len: c_int,
10903 );
10904 pub fn atg_mkldnn_rnn_layer(
10905 out__: *mut *mut C_tensor,
10906 input_: *mut C_tensor,
10907 weight0_: *mut C_tensor,
10908 weight1_: *mut C_tensor,
10909 weight2_: *mut C_tensor,
10910 weight3_: *mut C_tensor,
10911 hx__: *mut C_tensor,
10912 cx__: *mut C_tensor,
10913 reverse_: c_int,
10914 batch_sizes_data: *const i64,
10915 batch_sizes_len: c_int,
10916 mode_: i64,
10917 hidden_size_: i64,
10918 num_layers_: i64,
10919 has_biases_: c_int,
10920 bidirectional_: c_int,
10921 batch_first_: c_int,
10922 train_: c_int,
10923 );
10924 pub fn atg_mkldnn_rnn_layer_backward(
10925 out__: *mut *mut C_tensor,
10926 input_: *mut C_tensor,
10927 weight1_: *mut C_tensor,
10928 weight2_: *mut C_tensor,
10929 weight3_: *mut C_tensor,
10930 weight4_: *mut C_tensor,
10931 hx__: *mut C_tensor,
10932 cx_tmp_: *mut C_tensor,
10933 output_: *mut C_tensor,
10934 hy__: *mut C_tensor,
10935 cy__: *mut C_tensor,
10936 grad_output_: *mut C_tensor,
10937 grad_hy_: *mut C_tensor,
10938 grad_cy_: *mut C_tensor,
10939 reverse_: c_int,
10940 mode_: i64,
10941 hidden_size_: i64,
10942 num_layers_: i64,
10943 has_biases_: c_int,
10944 train_: c_int,
10945 bidirectional_: c_int,
10946 batch_sizes_data: *const i64,
10947 batch_sizes_len: c_int,
10948 batch_first_: c_int,
10949 workspace_: *mut C_tensor,
10950 );
10951 pub fn atg_mkldnn_rnn_layer_backward_out(
10952 out__: *mut *mut C_tensor,
10953 out0_: *mut C_tensor,
10954 out1_: *mut C_tensor,
10955 out2_: *mut C_tensor,
10956 out3_: *mut C_tensor,
10957 out4_: *mut C_tensor,
10958 out5_: *mut C_tensor,
10959 out6_: *mut C_tensor,
10960 input_: *mut C_tensor,
10961 weight1_: *mut C_tensor,
10962 weight2_: *mut C_tensor,
10963 weight3_: *mut C_tensor,
10964 weight4_: *mut C_tensor,
10965 hx__: *mut C_tensor,
10966 cx_tmp_: *mut C_tensor,
10967 output_: *mut C_tensor,
10968 hy__: *mut C_tensor,
10969 cy__: *mut C_tensor,
10970 grad_output_: *mut C_tensor,
10971 grad_hy_: *mut C_tensor,
10972 grad_cy_: *mut C_tensor,
10973 reverse_: c_int,
10974 mode_: i64,
10975 hidden_size_: i64,
10976 num_layers_: i64,
10977 has_biases_: c_int,
10978 train_: c_int,
10979 bidirectional_: c_int,
10980 batch_sizes_data: *const i64,
10981 batch_sizes_len: c_int,
10982 batch_first_: c_int,
10983 workspace_: *mut C_tensor,
10984 );
10985 pub fn atg_mkldnn_rnn_layer_out(
10986 out__: *mut *mut C_tensor,
10987 out0_: *mut C_tensor,
10988 out1_: *mut C_tensor,
10989 out2_: *mut C_tensor,
10990 out3_: *mut C_tensor,
10991 input_: *mut C_tensor,
10992 weight0_: *mut C_tensor,
10993 weight1_: *mut C_tensor,
10994 weight2_: *mut C_tensor,
10995 weight3_: *mut C_tensor,
10996 hx__: *mut C_tensor,
10997 cx__: *mut C_tensor,
10998 reverse_: c_int,
10999 batch_sizes_data: *const i64,
11000 batch_sizes_len: c_int,
11001 mode_: i64,
11002 hidden_size_: i64,
11003 num_layers_: i64,
11004 has_biases_: c_int,
11005 bidirectional_: c_int,
11006 batch_first_: c_int,
11007 train_: c_int,
11008 );
11009 pub fn atg_mm(out__: *mut *mut C_tensor, self_: *mut C_tensor, mat2_: *mut C_tensor);
11010 pub fn atg_mm_dtype(
11011 out__: *mut *mut C_tensor,
11012 self_: *mut C_tensor,
11013 mat2_: *mut C_tensor,
11014 out_dtype_: c_int,
11015 );
11016 pub fn atg_mm_dtype_out(
11017 out__: *mut *mut C_tensor,
11018 out_: *mut C_tensor,
11019 self_: *mut C_tensor,
11020 mat2_: *mut C_tensor,
11021 out_dtype_: c_int,
11022 );
11023 pub fn atg_mm_out(
11024 out__: *mut *mut C_tensor,
11025 out_: *mut C_tensor,
11026 self_: *mut C_tensor,
11027 mat2_: *mut C_tensor,
11028 );
11029 pub fn atg_mode(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, keepdim_: c_int);
11030 pub fn atg_mode_values(
11031 out__: *mut *mut C_tensor,
11032 values_: *mut C_tensor,
11033 indices_: *mut C_tensor,
11034 self_: *mut C_tensor,
11035 dim_: i64,
11036 keepdim_: c_int,
11037 );
11038 pub fn atg_moveaxis(
11039 out__: *mut *mut C_tensor,
11040 self_: *mut C_tensor,
11041 source_data: *const i64,
11042 source_len: c_int,
11043 destination_data: *const i64,
11044 destination_len: c_int,
11045 );
11046 pub fn atg_moveaxis_int(
11047 out__: *mut *mut C_tensor,
11048 self_: *mut C_tensor,
11049 source_: i64,
11050 destination_: i64,
11051 );
11052 pub fn atg_movedim(
11053 out__: *mut *mut C_tensor,
11054 self_: *mut C_tensor,
11055 source_data: *const i64,
11056 source_len: c_int,
11057 destination_data: *const i64,
11058 destination_len: c_int,
11059 );
11060 pub fn atg_movedim_int(
11061 out__: *mut *mut C_tensor,
11062 self_: *mut C_tensor,
11063 source_: i64,
11064 destination_: i64,
11065 );
11066 pub fn atg_mse_loss(
11067 out__: *mut *mut C_tensor,
11068 self_: *mut C_tensor,
11069 target_: *mut C_tensor,
11070 reduction_: i64,
11071 );
11072 pub fn atg_mse_loss_backward(
11073 out__: *mut *mut C_tensor,
11074 grad_output_: *mut C_tensor,
11075 self_: *mut C_tensor,
11076 target_: *mut C_tensor,
11077 reduction_: i64,
11078 );
11079 pub fn atg_mse_loss_backward_grad_input(
11080 out__: *mut *mut C_tensor,
11081 grad_input_: *mut C_tensor,
11082 grad_output_: *mut C_tensor,
11083 self_: *mut C_tensor,
11084 target_: *mut C_tensor,
11085 reduction_: i64,
11086 );
11087 pub fn atg_mse_loss_out(
11088 out__: *mut *mut C_tensor,
11089 out_: *mut C_tensor,
11090 self_: *mut C_tensor,
11091 target_: *mut C_tensor,
11092 reduction_: i64,
11093 );
11094 pub fn atg_msort(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11095 pub fn atg_msort_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
11096 pub fn atg_mt(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11097 pub fn atg_mul(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
11098 pub fn atg_mul_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
11099 pub fn atg_mul_out(
11100 out__: *mut *mut C_tensor,
11101 out_: *mut C_tensor,
11102 self_: *mut C_tensor,
11103 other_: *mut C_tensor,
11104 );
11105 pub fn atg_mul_scalar(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
11106 pub fn atg_mul_scalar_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
11107 pub fn atg_mul_scalar_out(
11108 out__: *mut *mut C_tensor,
11109 out_: *mut C_tensor,
11110 self_: *mut C_tensor,
11111 other_: *mut C_scalar,
11112 );
11113 pub fn atg_multi_margin_loss_backward(
11114 out__: *mut *mut C_tensor,
11115 grad_output_: *mut C_tensor,
11116 self_: *mut C_tensor,
11117 target_: *mut C_tensor,
11118 p_: *mut C_scalar,
11119 margin_: *mut C_scalar,
11120 weight_: *mut C_tensor,
11121 reduction_: i64,
11122 );
11123 pub fn atg_multi_margin_loss_backward_grad_input(
11124 out__: *mut *mut C_tensor,
11125 grad_input_: *mut C_tensor,
11126 grad_output_: *mut C_tensor,
11127 self_: *mut C_tensor,
11128 target_: *mut C_tensor,
11129 p_: *mut C_scalar,
11130 margin_: *mut C_scalar,
11131 weight_: *mut C_tensor,
11132 reduction_: i64,
11133 );
11134 pub fn atg_multilabel_margin_loss(
11135 out__: *mut *mut C_tensor,
11136 self_: *mut C_tensor,
11137 target_: *mut C_tensor,
11138 reduction_: i64,
11139 );
11140 pub fn atg_multilabel_margin_loss_backward(
11141 out__: *mut *mut C_tensor,
11142 grad_output_: *mut C_tensor,
11143 self_: *mut C_tensor,
11144 target_: *mut C_tensor,
11145 reduction_: i64,
11146 is_target_: *mut C_tensor,
11147 );
11148 pub fn atg_multilabel_margin_loss_backward_grad_input(
11149 out__: *mut *mut C_tensor,
11150 grad_input_: *mut C_tensor,
11151 grad_output_: *mut C_tensor,
11152 self_: *mut C_tensor,
11153 target_: *mut C_tensor,
11154 reduction_: i64,
11155 is_target_: *mut C_tensor,
11156 );
11157 pub fn atg_multilabel_margin_loss_out(
11158 out__: *mut *mut C_tensor,
11159 out_: *mut C_tensor,
11160 self_: *mut C_tensor,
11161 target_: *mut C_tensor,
11162 reduction_: i64,
11163 );
11164 pub fn atg_multinomial(
11165 out__: *mut *mut C_tensor,
11166 self_: *mut C_tensor,
11167 num_samples_: i64,
11168 replacement_: c_int,
11169 );
11170 pub fn atg_multinomial_out(
11171 out__: *mut *mut C_tensor,
11172 out_: *mut C_tensor,
11173 self_: *mut C_tensor,
11174 num_samples_: i64,
11175 replacement_: c_int,
11176 );
11177 pub fn atg_multiply(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
11178 pub fn atg_multiply_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
11179 pub fn atg_multiply_out(
11180 out__: *mut *mut C_tensor,
11181 out_: *mut C_tensor,
11182 self_: *mut C_tensor,
11183 other_: *mut C_tensor,
11184 );
11185 pub fn atg_multiply_scalar(
11186 out__: *mut *mut C_tensor,
11187 self_: *mut C_tensor,
11188 other_: *mut C_scalar,
11189 );
11190 pub fn atg_multiply_scalar_(
11191 out__: *mut *mut C_tensor,
11192 self_: *mut C_tensor,
11193 other_: *mut C_scalar,
11194 );
11195 pub fn atg_mv(out__: *mut *mut C_tensor, self_: *mut C_tensor, vec_: *mut C_tensor);
11196 pub fn atg_mv_out(
11197 out__: *mut *mut C_tensor,
11198 out_: *mut C_tensor,
11199 self_: *mut C_tensor,
11200 vec_: *mut C_tensor,
11201 );
11202 pub fn atg_mvlgamma(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: i64);
11203 pub fn atg_mvlgamma_(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: i64);
11204 pub fn atg_mvlgamma_out(
11205 out__: *mut *mut C_tensor,
11206 out_: *mut C_tensor,
11207 self_: *mut C_tensor,
11208 p_: i64,
11209 );
11210 pub fn atg_nan_to_num(
11211 out__: *mut *mut C_tensor,
11212 self_: *mut C_tensor,
11213 nan_v: f64,
11214 nan_null: i8,
11215 posinf_v: f64,
11216 posinf_null: i8,
11217 neginf_v: f64,
11218 neginf_null: i8,
11219 );
11220 pub fn atg_nan_to_num_(
11221 out__: *mut *mut C_tensor,
11222 self_: *mut C_tensor,
11223 nan_v: f64,
11224 nan_null: i8,
11225 posinf_v: f64,
11226 posinf_null: i8,
11227 neginf_v: f64,
11228 neginf_null: i8,
11229 );
11230 pub fn atg_nan_to_num_out(
11231 out__: *mut *mut C_tensor,
11232 out_: *mut C_tensor,
11233 self_: *mut C_tensor,
11234 nan_v: f64,
11235 nan_null: i8,
11236 posinf_v: f64,
11237 posinf_null: i8,
11238 neginf_v: f64,
11239 neginf_null: i8,
11240 );
11241 pub fn atg_nanmean(
11242 out__: *mut *mut C_tensor,
11243 self_: *mut C_tensor,
11244 dim_data: *const i64,
11245 dim_len: c_int,
11246 keepdim_: c_int,
11247 dtype_: c_int,
11248 );
11249 pub fn atg_nanmean_out(
11250 out__: *mut *mut C_tensor,
11251 out_: *mut C_tensor,
11252 self_: *mut C_tensor,
11253 dim_data: *const i64,
11254 dim_len: c_int,
11255 keepdim_: c_int,
11256 dtype_: c_int,
11257 );
11258 pub fn atg_nanmedian(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11259 pub fn atg_nanmedian_dim(
11260 out__: *mut *mut C_tensor,
11261 self_: *mut C_tensor,
11262 dim_: i64,
11263 keepdim_: c_int,
11264 );
11265 pub fn atg_nanmedian_dim_values(
11266 out__: *mut *mut C_tensor,
11267 values_: *mut C_tensor,
11268 indices_: *mut C_tensor,
11269 self_: *mut C_tensor,
11270 dim_: i64,
11271 keepdim_: c_int,
11272 );
11273 pub fn atg_nanmedian_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
11274 pub fn atg_nanquantile(
11275 out__: *mut *mut C_tensor,
11276 self_: *mut C_tensor,
11277 q_: *mut C_tensor,
11278 dim_v: i64,
11279 dim_null: i8,
11280 keepdim_: c_int,
11281 interpolation_ptr: *const u8,
11282 interpolation_len: c_int,
11283 );
11284 pub fn atg_nanquantile_out(
11285 out__: *mut *mut C_tensor,
11286 out_: *mut C_tensor,
11287 self_: *mut C_tensor,
11288 q_: *mut C_tensor,
11289 dim_v: i64,
11290 dim_null: i8,
11291 keepdim_: c_int,
11292 interpolation_ptr: *const u8,
11293 interpolation_len: c_int,
11294 );
11295 pub fn atg_nanquantile_scalar(
11296 out__: *mut *mut C_tensor,
11297 self_: *mut C_tensor,
11298 q_: f64,
11299 dim_v: i64,
11300 dim_null: i8,
11301 keepdim_: c_int,
11302 interpolation_ptr: *const u8,
11303 interpolation_len: c_int,
11304 );
11305 pub fn atg_nanquantile_scalar_out(
11306 out__: *mut *mut C_tensor,
11307 out_: *mut C_tensor,
11308 self_: *mut C_tensor,
11309 q_: f64,
11310 dim_v: i64,
11311 dim_null: i8,
11312 keepdim_: c_int,
11313 interpolation_ptr: *const u8,
11314 interpolation_len: c_int,
11315 );
11316 pub fn atg_nansum(
11317 out__: *mut *mut C_tensor,
11318 self_: *mut C_tensor,
11319 dim_data: *const i64,
11320 dim_len: c_int,
11321 keepdim_: c_int,
11322 dtype_: c_int,
11323 );
11324 pub fn atg_nansum_out(
11325 out__: *mut *mut C_tensor,
11326 out_: *mut C_tensor,
11327 self_: *mut C_tensor,
11328 dim_data: *const i64,
11329 dim_len: c_int,
11330 keepdim_: c_int,
11331 dtype_: c_int,
11332 );
11333 pub fn atg_narrow(
11334 out__: *mut *mut C_tensor,
11335 self_: *mut C_tensor,
11336 dim_: i64,
11337 start_: i64,
11338 length_: i64,
11339 );
11340 pub fn atg_narrow_copy(
11341 out__: *mut *mut C_tensor,
11342 self_: *mut C_tensor,
11343 dim_: i64,
11344 start_: i64,
11345 length_: i64,
11346 );
11347 pub fn atg_narrow_copy_out(
11348 out__: *mut *mut C_tensor,
11349 out_: *mut C_tensor,
11350 self_: *mut C_tensor,
11351 dim_: i64,
11352 start_: i64,
11353 length_: i64,
11354 );
11355 pub fn atg_narrow_tensor(
11356 out__: *mut *mut C_tensor,
11357 self_: *mut C_tensor,
11358 dim_: i64,
11359 start_: *mut C_tensor,
11360 length_: i64,
11361 );
11362 pub fn atg_native_batch_norm(
11363 out__: *mut *mut C_tensor,
11364 input_: *mut C_tensor,
11365 weight_: *mut C_tensor,
11366 bias_: *mut C_tensor,
11367 running_mean_: *mut C_tensor,
11368 running_var_: *mut C_tensor,
11369 training_: c_int,
11370 momentum_: f64,
11371 eps_: f64,
11372 );
11373 pub fn atg_native_batch_norm_out(
11374 out__: *mut *mut C_tensor,
11375 out_: *mut C_tensor,
11376 save_mean_: *mut C_tensor,
11377 save_invstd_: *mut C_tensor,
11378 input_: *mut C_tensor,
11379 weight_: *mut C_tensor,
11380 bias_: *mut C_tensor,
11381 running_mean_: *mut C_tensor,
11382 running_var_: *mut C_tensor,
11383 training_: c_int,
11384 momentum_: f64,
11385 eps_: f64,
11386 );
11387 pub fn atg_native_channel_shuffle(
11388 out__: *mut *mut C_tensor,
11389 self_: *mut C_tensor,
11390 groups_: i64,
11391 );
11392 pub fn atg_native_dropout(
11393 out__: *mut *mut C_tensor,
11394 input_: *mut C_tensor,
11395 p_: f64,
11396 train_: c_int,
11397 );
11398 pub fn atg_native_dropout_backward(
11399 out__: *mut *mut C_tensor,
11400 grad_output_: *mut C_tensor,
11401 mask_: *mut C_tensor,
11402 scale_: f64,
11403 );
11404 pub fn atg_native_dropout_backward_out(
11405 out__: *mut *mut C_tensor,
11406 out_: *mut C_tensor,
11407 grad_output_: *mut C_tensor,
11408 mask_: *mut C_tensor,
11409 scale_: f64,
11410 );
11411 pub fn atg_native_dropout_out(
11412 out__: *mut *mut C_tensor,
11413 out0_: *mut C_tensor,
11414 out1_: *mut C_tensor,
11415 input_: *mut C_tensor,
11416 p_: f64,
11417 train_: c_int,
11418 );
11419 pub fn atg_native_group_norm(
11420 out__: *mut *mut C_tensor,
11421 input_: *mut C_tensor,
11422 weight_: *mut C_tensor,
11423 bias_: *mut C_tensor,
11424 n_: i64,
11425 C_: i64,
11426 HxW_: i64,
11427 group_: i64,
11428 eps_: f64,
11429 );
11430 pub fn atg_native_group_norm_out(
11431 out__: *mut *mut C_tensor,
11432 out0_: *mut C_tensor,
11433 out1_: *mut C_tensor,
11434 out2_: *mut C_tensor,
11435 input_: *mut C_tensor,
11436 weight_: *mut C_tensor,
11437 bias_: *mut C_tensor,
11438 n_: i64,
11439 C_: i64,
11440 HxW_: i64,
11441 group_: i64,
11442 eps_: f64,
11443 );
11444 pub fn atg_native_layer_norm(
11445 out__: *mut *mut C_tensor,
11446 input_: *mut C_tensor,
11447 normalized_shape_data: *const i64,
11448 normalized_shape_len: c_int,
11449 weight_: *mut C_tensor,
11450 bias_: *mut C_tensor,
11451 eps_: f64,
11452 );
11453 pub fn atg_native_layer_norm_out(
11454 out__: *mut *mut C_tensor,
11455 out0_: *mut C_tensor,
11456 out1_: *mut C_tensor,
11457 out2_: *mut C_tensor,
11458 input_: *mut C_tensor,
11459 normalized_shape_data: *const i64,
11460 normalized_shape_len: c_int,
11461 weight_: *mut C_tensor,
11462 bias_: *mut C_tensor,
11463 eps_: f64,
11464 );
11465 pub fn atg_native_norm(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11466 pub fn atg_native_norm_out(
11467 out__: *mut *mut C_tensor,
11468 out_: *mut C_tensor,
11469 self_: *mut C_tensor,
11470 );
11471 pub fn atg_native_norm_scalaropt_dim_dtype(
11472 out__: *mut *mut C_tensor,
11473 self_: *mut C_tensor,
11474 p_: *mut C_scalar,
11475 dim_data: *const i64,
11476 dim_len: c_int,
11477 keepdim_: c_int,
11478 dtype_: c_int,
11479 );
11480 pub fn atg_native_norm_scalaropt_dim_dtype_out(
11481 out__: *mut *mut C_tensor,
11482 out_: *mut C_tensor,
11483 self_: *mut C_tensor,
11484 p_: *mut C_scalar,
11485 dim_data: *const i64,
11486 dim_len: c_int,
11487 keepdim_: c_int,
11488 dtype_: c_int,
11489 );
11490 pub fn atg_ne(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
11491 pub fn atg_ne_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
11492 pub fn atg_ne_scalar_out(
11493 out__: *mut *mut C_tensor,
11494 out_: *mut C_tensor,
11495 self_: *mut C_tensor,
11496 other_: *mut C_scalar,
11497 );
11498 pub fn atg_ne_tensor(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
11499 pub fn atg_ne_tensor_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
11500 pub fn atg_ne_tensor_out(
11501 out__: *mut *mut C_tensor,
11502 out_: *mut C_tensor,
11503 self_: *mut C_tensor,
11504 other_: *mut C_tensor,
11505 );
11506 pub fn atg_neg(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11507 pub fn atg_neg_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11508 pub fn atg_neg_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
11509 pub fn atg_negative(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11510 pub fn atg_negative_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11511 pub fn atg_negative_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
11512 pub fn atg_nested_to_padded_tensor(
11513 out__: *mut *mut C_tensor,
11514 self_: *mut C_tensor,
11515 padding_: f64,
11516 output_size_data: *const i64,
11517 output_size_len: c_int,
11518 );
11519 pub fn atg_new_empty(
11520 out__: *mut *mut C_tensor,
11521 self_: *mut C_tensor,
11522 size_data: *const i64,
11523 size_len: c_int,
11524 options_kind: c_int,
11525 options_device: c_int,
11526 );
11527 pub fn atg_new_empty_out(
11528 out__: *mut *mut C_tensor,
11529 out_: *mut C_tensor,
11530 self_: *mut C_tensor,
11531 size_data: *const i64,
11532 size_len: c_int,
11533 );
11534 pub fn atg_new_empty_strided(
11535 out__: *mut *mut C_tensor,
11536 self_: *mut C_tensor,
11537 size_data: *const i64,
11538 size_len: c_int,
11539 stride_data: *const i64,
11540 stride_len: c_int,
11541 options_kind: c_int,
11542 options_device: c_int,
11543 );
11544 pub fn atg_new_empty_strided_out(
11545 out__: *mut *mut C_tensor,
11546 out_: *mut C_tensor,
11547 self_: *mut C_tensor,
11548 size_data: *const i64,
11549 size_len: c_int,
11550 stride_data: *const i64,
11551 stride_len: c_int,
11552 );
11553 pub fn atg_new_full(
11554 out__: *mut *mut C_tensor,
11555 self_: *mut C_tensor,
11556 size_data: *const i64,
11557 size_len: c_int,
11558 fill_value_: *mut C_scalar,
11559 options_kind: c_int,
11560 options_device: c_int,
11561 );
11562 pub fn atg_new_full_out(
11563 out__: *mut *mut C_tensor,
11564 out_: *mut C_tensor,
11565 self_: *mut C_tensor,
11566 size_data: *const i64,
11567 size_len: c_int,
11568 fill_value_: *mut C_scalar,
11569 );
11570 pub fn atg_new_ones(
11571 out__: *mut *mut C_tensor,
11572 self_: *mut C_tensor,
11573 size_data: *const i64,
11574 size_len: c_int,
11575 options_kind: c_int,
11576 options_device: c_int,
11577 );
11578 pub fn atg_new_ones_out(
11579 out__: *mut *mut C_tensor,
11580 out_: *mut C_tensor,
11581 self_: *mut C_tensor,
11582 size_data: *const i64,
11583 size_len: c_int,
11584 );
11585 pub fn atg_new_zeros(
11586 out__: *mut *mut C_tensor,
11587 self_: *mut C_tensor,
11588 size_data: *const i64,
11589 size_len: c_int,
11590 options_kind: c_int,
11591 options_device: c_int,
11592 );
11593 pub fn atg_new_zeros_out(
11594 out__: *mut *mut C_tensor,
11595 out_: *mut C_tensor,
11596 self_: *mut C_tensor,
11597 size_data: *const i64,
11598 size_len: c_int,
11599 );
11600 pub fn atg_nextafter(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
11601 pub fn atg_nextafter_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
11602 pub fn atg_nextafter_out(
11603 out__: *mut *mut C_tensor,
11604 out_: *mut C_tensor,
11605 self_: *mut C_tensor,
11606 other_: *mut C_tensor,
11607 );
11608 pub fn atg_nll_loss(
11609 out__: *mut *mut C_tensor,
11610 self_: *mut C_tensor,
11611 target_: *mut C_tensor,
11612 weight_: *mut C_tensor,
11613 reduction_: i64,
11614 ignore_index_: i64,
11615 );
11616 pub fn atg_nll_loss2d(
11617 out__: *mut *mut C_tensor,
11618 self_: *mut C_tensor,
11619 target_: *mut C_tensor,
11620 weight_: *mut C_tensor,
11621 reduction_: i64,
11622 ignore_index_: i64,
11623 );
11624 pub fn atg_nll_loss2d_backward(
11625 out__: *mut *mut C_tensor,
11626 grad_output_: *mut C_tensor,
11627 self_: *mut C_tensor,
11628 target_: *mut C_tensor,
11629 weight_: *mut C_tensor,
11630 reduction_: i64,
11631 ignore_index_: i64,
11632 total_weight_: *mut C_tensor,
11633 );
11634 pub fn atg_nll_loss2d_backward_grad_input(
11635 out__: *mut *mut C_tensor,
11636 grad_input_: *mut C_tensor,
11637 grad_output_: *mut C_tensor,
11638 self_: *mut C_tensor,
11639 target_: *mut C_tensor,
11640 weight_: *mut C_tensor,
11641 reduction_: i64,
11642 ignore_index_: i64,
11643 total_weight_: *mut C_tensor,
11644 );
11645 pub fn atg_nll_loss2d_out(
11646 out__: *mut *mut C_tensor,
11647 out_: *mut C_tensor,
11648 self_: *mut C_tensor,
11649 target_: *mut C_tensor,
11650 weight_: *mut C_tensor,
11651 reduction_: i64,
11652 ignore_index_: i64,
11653 );
11654 pub fn atg_nll_loss_backward(
11655 out__: *mut *mut C_tensor,
11656 grad_output_: *mut C_tensor,
11657 self_: *mut C_tensor,
11658 target_: *mut C_tensor,
11659 weight_: *mut C_tensor,
11660 reduction_: i64,
11661 ignore_index_: i64,
11662 total_weight_: *mut C_tensor,
11663 );
11664 pub fn atg_nll_loss_backward_grad_input(
11665 out__: *mut *mut C_tensor,
11666 grad_input_: *mut C_tensor,
11667 grad_output_: *mut C_tensor,
11668 self_: *mut C_tensor,
11669 target_: *mut C_tensor,
11670 weight_: *mut C_tensor,
11671 reduction_: i64,
11672 ignore_index_: i64,
11673 total_weight_: *mut C_tensor,
11674 );
11675 pub fn atg_nll_loss_nd(
11676 out__: *mut *mut C_tensor,
11677 self_: *mut C_tensor,
11678 target_: *mut C_tensor,
11679 weight_: *mut C_tensor,
11680 reduction_: i64,
11681 ignore_index_: i64,
11682 );
11683 pub fn atg_nll_loss_out(
11684 out__: *mut *mut C_tensor,
11685 out_: *mut C_tensor,
11686 self_: *mut C_tensor,
11687 target_: *mut C_tensor,
11688 weight_: *mut C_tensor,
11689 reduction_: i64,
11690 ignore_index_: i64,
11691 );
11692 pub fn atg_nonzero(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11693 pub fn atg_nonzero_numpy(self_: *mut C_tensor) -> *mut *mut C_tensor;
11694 pub fn atg_nonzero_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
11695 pub fn atg_nonzero_static(
11696 out__: *mut *mut C_tensor,
11697 self_: *mut C_tensor,
11698 size_: i64,
11699 fill_value_: i64,
11700 );
11701 pub fn atg_nonzero_static_out(
11702 out__: *mut *mut C_tensor,
11703 out_: *mut C_tensor,
11704 self_: *mut C_tensor,
11705 size_: i64,
11706 fill_value_: i64,
11707 );
11708 pub fn atg_norm(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11709 pub fn atg_norm_dtype_out(
11710 out__: *mut *mut C_tensor,
11711 out_: *mut C_tensor,
11712 self_: *mut C_tensor,
11713 p_: *mut C_scalar,
11714 dim_data: *const i64,
11715 dim_len: c_int,
11716 keepdim_: c_int,
11717 dtype_: c_int,
11718 );
11719 pub fn atg_norm_except_dim(out__: *mut *mut C_tensor, v_: *mut C_tensor, pow_: i64, dim_: i64);
11720 pub fn atg_norm_out(
11721 out__: *mut *mut C_tensor,
11722 out_: *mut C_tensor,
11723 self_: *mut C_tensor,
11724 p_: *mut C_scalar,
11725 dim_data: *const i64,
11726 dim_len: c_int,
11727 keepdim_: c_int,
11728 );
11729 pub fn atg_norm_scalar_out(
11730 out__: *mut *mut C_tensor,
11731 out_: *mut C_tensor,
11732 self_: *mut C_tensor,
11733 );
11734 pub fn atg_norm_scalaropt_dim(
11735 out__: *mut *mut C_tensor,
11736 self_: *mut C_tensor,
11737 p_: *mut C_scalar,
11738 dim_data: *const i64,
11739 dim_len: c_int,
11740 keepdim_: c_int,
11741 );
11742 pub fn atg_norm_scalaropt_dim_dtype(
11743 out__: *mut *mut C_tensor,
11744 self_: *mut C_tensor,
11745 p_: *mut C_scalar,
11746 dim_data: *const i64,
11747 dim_len: c_int,
11748 keepdim_: c_int,
11749 dtype_: c_int,
11750 );
11751 pub fn atg_norm_scalaropt_dtype(
11752 out__: *mut *mut C_tensor,
11753 self_: *mut C_tensor,
11754 p_: *mut C_scalar,
11755 dtype_: c_int,
11756 );
11757 pub fn atg_norm_scalaropt_dtype_out(
11758 out__: *mut *mut C_tensor,
11759 out_: *mut C_tensor,
11760 self_: *mut C_tensor,
11761 p_: *mut C_scalar,
11762 dtype_: c_int,
11763 );
11764 pub fn atg_normal_(out__: *mut *mut C_tensor, self_: *mut C_tensor, mean_: f64, std_: f64);
11765 pub fn atg_normal_functional(
11766 out__: *mut *mut C_tensor,
11767 self_: *mut C_tensor,
11768 mean_: f64,
11769 std_: f64,
11770 );
11771 pub fn atg_not_equal(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
11772 pub fn atg_not_equal_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
11773 pub fn atg_not_equal_scalar_out(
11774 out__: *mut *mut C_tensor,
11775 out_: *mut C_tensor,
11776 self_: *mut C_tensor,
11777 other_: *mut C_scalar,
11778 );
11779 pub fn atg_not_equal_tensor(
11780 out__: *mut *mut C_tensor,
11781 self_: *mut C_tensor,
11782 other_: *mut C_tensor,
11783 );
11784 pub fn atg_not_equal_tensor_(
11785 out__: *mut *mut C_tensor,
11786 self_: *mut C_tensor,
11787 other_: *mut C_tensor,
11788 );
11789 pub fn atg_not_equal_tensor_out(
11790 out__: *mut *mut C_tensor,
11791 out_: *mut C_tensor,
11792 self_: *mut C_tensor,
11793 other_: *mut C_tensor,
11794 );
11795 pub fn atg_nuclear_norm(out__: *mut *mut C_tensor, self_: *mut C_tensor, keepdim_: c_int);
11796 pub fn atg_nuclear_norm_dim(
11797 out__: *mut *mut C_tensor,
11798 self_: *mut C_tensor,
11799 dim_data: *const i64,
11800 dim_len: c_int,
11801 keepdim_: c_int,
11802 );
11803 pub fn atg_nuclear_norm_dim_out(
11804 out__: *mut *mut C_tensor,
11805 out_: *mut C_tensor,
11806 self_: *mut C_tensor,
11807 dim_data: *const i64,
11808 dim_len: c_int,
11809 keepdim_: c_int,
11810 );
11811 pub fn atg_nuclear_norm_out(
11812 out__: *mut *mut C_tensor,
11813 out_: *mut C_tensor,
11814 self_: *mut C_tensor,
11815 keepdim_: c_int,
11816 );
11817 pub fn atg_numpy_t(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11818 pub fn atg_one_hot(out__: *mut *mut C_tensor, self_: *mut C_tensor, num_classes_: i64);
11819 pub fn atg_ones(
11820 out__: *mut *mut C_tensor,
11821 size_data: *const i64,
11822 size_len: c_int,
11823 options_kind: c_int,
11824 options_device: c_int,
11825 );
11826 pub fn atg_ones_like(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11827 pub fn atg_ones_like_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
11828 pub fn atg_ones_out(
11829 out__: *mut *mut C_tensor,
11830 out_: *mut C_tensor,
11831 size_data: *const i64,
11832 size_len: c_int,
11833 );
11834 pub fn atg_orgqr(out__: *mut *mut C_tensor, self_: *mut C_tensor, input2_: *mut C_tensor);
11835 pub fn atg_orgqr_out(
11836 out__: *mut *mut C_tensor,
11837 out_: *mut C_tensor,
11838 self_: *mut C_tensor,
11839 input2_: *mut C_tensor,
11840 );
11841 pub fn atg_ormqr(
11842 out__: *mut *mut C_tensor,
11843 self_: *mut C_tensor,
11844 input2_: *mut C_tensor,
11845 input3_: *mut C_tensor,
11846 left_: c_int,
11847 transpose_: c_int,
11848 );
11849 pub fn atg_ormqr_out(
11850 out__: *mut *mut C_tensor,
11851 out_: *mut C_tensor,
11852 self_: *mut C_tensor,
11853 input2_: *mut C_tensor,
11854 input3_: *mut C_tensor,
11855 left_: c_int,
11856 transpose_: c_int,
11857 );
11858 pub fn atg_outer(out__: *mut *mut C_tensor, self_: *mut C_tensor, vec2_: *mut C_tensor);
11859 pub fn atg_outer_out(
11860 out__: *mut *mut C_tensor,
11861 out_: *mut C_tensor,
11862 self_: *mut C_tensor,
11863 vec2_: *mut C_tensor,
11864 );
11865 pub fn atg_output_nr(self_: *mut C_tensor) -> i64;
11866 pub fn atg_pad(
11867 out__: *mut *mut C_tensor,
11868 self_: *mut C_tensor,
11869 pad_data: *const i64,
11870 pad_len: c_int,
11871 mode_ptr: *const u8,
11872 mode_len: c_int,
11873 value_v: f64,
11874 value_null: i8,
11875 );
11876 pub fn atg_pad_sequence(
11877 out__: *mut *mut C_tensor,
11878 sequences_data: *const *mut C_tensor,
11879 sequences_len: c_int,
11880 batch_first_: c_int,
11881 padding_value_: f64,
11882 padding_side_ptr: *const u8,
11883 padding_side_len: c_int,
11884 );
11885 pub fn atg_pairwise_distance(
11886 out__: *mut *mut C_tensor,
11887 x1_: *mut C_tensor,
11888 x2_: *mut C_tensor,
11889 p_: f64,
11890 eps_: f64,
11891 keepdim_: c_int,
11892 );
11893 pub fn atg_pdist(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: f64);
11894 pub fn atg_permute(
11895 out__: *mut *mut C_tensor,
11896 self_: *mut C_tensor,
11897 dims_data: *const i64,
11898 dims_len: c_int,
11899 );
11900 pub fn atg_permute_copy(
11901 out__: *mut *mut C_tensor,
11902 self_: *mut C_tensor,
11903 dims_data: *const i64,
11904 dims_len: c_int,
11905 );
11906 pub fn atg_permute_copy_out(
11907 out__: *mut *mut C_tensor,
11908 out_: *mut C_tensor,
11909 self_: *mut C_tensor,
11910 dims_data: *const i64,
11911 dims_len: c_int,
11912 );
11913 pub fn atg_pin_memory(out__: *mut *mut C_tensor, self_: *mut C_tensor, device_: c_int);
11914 pub fn atg_pinverse(out__: *mut *mut C_tensor, self_: *mut C_tensor, rcond_: f64);
11915 pub fn atg_pixel_shuffle(out__: *mut *mut C_tensor, self_: *mut C_tensor, upscale_factor_: i64);
11916 pub fn atg_pixel_shuffle_out(
11917 out__: *mut *mut C_tensor,
11918 out_: *mut C_tensor,
11919 self_: *mut C_tensor,
11920 upscale_factor_: i64,
11921 );
11922 pub fn atg_pixel_unshuffle(
11923 out__: *mut *mut C_tensor,
11924 self_: *mut C_tensor,
11925 downscale_factor_: i64,
11926 );
11927 pub fn atg_pixel_unshuffle_out(
11928 out__: *mut *mut C_tensor,
11929 out_: *mut C_tensor,
11930 self_: *mut C_tensor,
11931 downscale_factor_: i64,
11932 );
11933 pub fn atg_poisson(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11934 pub fn atg_poisson_nll_loss(
11935 out__: *mut *mut C_tensor,
11936 input_: *mut C_tensor,
11937 target_: *mut C_tensor,
11938 log_input_: c_int,
11939 full_: c_int,
11940 eps_: f64,
11941 reduction_: i64,
11942 );
11943 pub fn atg_poisson_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
11944 pub fn atg_polar(out__: *mut *mut C_tensor, abs_: *mut C_tensor, angle_: *mut C_tensor);
11945 pub fn atg_polar_out(
11946 out__: *mut *mut C_tensor,
11947 out_: *mut C_tensor,
11948 abs_: *mut C_tensor,
11949 angle_: *mut C_tensor,
11950 );
11951 pub fn atg_polygamma(out__: *mut *mut C_tensor, n_: i64, self_: *mut C_tensor);
11952 pub fn atg_polygamma_(out__: *mut *mut C_tensor, self_: *mut C_tensor, n_: i64);
11953 pub fn atg_polygamma_out(
11954 out__: *mut *mut C_tensor,
11955 out_: *mut C_tensor,
11956 n_: i64,
11957 self_: *mut C_tensor,
11958 );
11959 pub fn atg_positive(out__: *mut *mut C_tensor, self_: *mut C_tensor);
11960 pub fn atg_pow(out__: *mut *mut C_tensor, self_: *mut C_tensor, exponent_: *mut C_tensor);
11961 pub fn atg_pow_(out__: *mut *mut C_tensor, self_: *mut C_tensor, exponent_: *mut C_scalar);
11962 pub fn atg_pow_scalar(
11963 out__: *mut *mut C_tensor,
11964 self_scalar_: *mut C_scalar,
11965 exponent_: *mut C_tensor,
11966 );
11967 pub fn atg_pow_scalar_out(
11968 out__: *mut *mut C_tensor,
11969 out_: *mut C_tensor,
11970 self_scalar_: *mut C_scalar,
11971 exponent_: *mut C_tensor,
11972 );
11973 pub fn atg_pow_tensor_(
11974 out__: *mut *mut C_tensor,
11975 self_: *mut C_tensor,
11976 exponent_: *mut C_tensor,
11977 );
11978 pub fn atg_pow_tensor_scalar(
11979 out__: *mut *mut C_tensor,
11980 self_: *mut C_tensor,
11981 exponent_: *mut C_scalar,
11982 );
11983 pub fn atg_pow_tensor_scalar_out(
11984 out__: *mut *mut C_tensor,
11985 out_: *mut C_tensor,
11986 self_: *mut C_tensor,
11987 exponent_: *mut C_scalar,
11988 );
11989 pub fn atg_pow_tensor_tensor_out(
11990 out__: *mut *mut C_tensor,
11991 out_: *mut C_tensor,
11992 self_: *mut C_tensor,
11993 exponent_: *mut C_tensor,
11994 );
11995 pub fn atg_prelu(out__: *mut *mut C_tensor, self_: *mut C_tensor, weight_: *mut C_tensor);
11996 pub fn atg_prod(out__: *mut *mut C_tensor, self_: *mut C_tensor, dtype_: c_int);
11997 pub fn atg_prod_dim_int(
11998 out__: *mut *mut C_tensor,
11999 self_: *mut C_tensor,
12000 dim_: i64,
12001 keepdim_: c_int,
12002 dtype_: c_int,
12003 );
12004 pub fn atg_prod_int_out(
12005 out__: *mut *mut C_tensor,
12006 out_: *mut C_tensor,
12007 self_: *mut C_tensor,
12008 dim_: i64,
12009 keepdim_: c_int,
12010 dtype_: c_int,
12011 );
12012 pub fn atg_prod_out(
12013 out__: *mut *mut C_tensor,
12014 out_: *mut C_tensor,
12015 self_: *mut C_tensor,
12016 dtype_: c_int,
12017 );
12018 pub fn atg_put(
12019 out__: *mut *mut C_tensor,
12020 self_: *mut C_tensor,
12021 index_: *mut C_tensor,
12022 source_: *mut C_tensor,
12023 accumulate_: c_int,
12024 );
12025 pub fn atg_put_(
12026 out__: *mut *mut C_tensor,
12027 self_: *mut C_tensor,
12028 index_: *mut C_tensor,
12029 source_: *mut C_tensor,
12030 accumulate_: c_int,
12031 );
12032 pub fn atg_put_out(
12033 out__: *mut *mut C_tensor,
12034 out_: *mut C_tensor,
12035 self_: *mut C_tensor,
12036 index_: *mut C_tensor,
12037 source_: *mut C_tensor,
12038 accumulate_: c_int,
12039 );
12040 pub fn atg_q_per_channel_axis(self_: *mut C_tensor) -> i64;
12041 pub fn atg_q_per_channel_scales(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12042 pub fn atg_q_per_channel_scales_out(
12043 out__: *mut *mut C_tensor,
12044 out_: *mut C_tensor,
12045 self_: *mut C_tensor,
12046 );
12047 pub fn atg_q_per_channel_zero_points(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12048 pub fn atg_q_per_channel_zero_points_out(
12049 out__: *mut *mut C_tensor,
12050 out_: *mut C_tensor,
12051 self_: *mut C_tensor,
12052 );
12053 pub fn atg_q_scale(self_: *mut C_tensor) -> f64;
12054 pub fn atg_q_zero_point(self_: *mut C_tensor) -> i64;
12055 pub fn atg_qr(out__: *mut *mut C_tensor, self_: *mut C_tensor, some_: c_int);
12056 pub fn atg_qr_q(
12057 out__: *mut *mut C_tensor,
12058 Q_: *mut C_tensor,
12059 R_: *mut C_tensor,
12060 self_: *mut C_tensor,
12061 some_: c_int,
12062 );
12063 pub fn atg_quantile(
12064 out__: *mut *mut C_tensor,
12065 self_: *mut C_tensor,
12066 q_: *mut C_tensor,
12067 dim_v: i64,
12068 dim_null: i8,
12069 keepdim_: c_int,
12070 interpolation_ptr: *const u8,
12071 interpolation_len: c_int,
12072 );
12073 pub fn atg_quantile_out(
12074 out__: *mut *mut C_tensor,
12075 out_: *mut C_tensor,
12076 self_: *mut C_tensor,
12077 q_: *mut C_tensor,
12078 dim_v: i64,
12079 dim_null: i8,
12080 keepdim_: c_int,
12081 interpolation_ptr: *const u8,
12082 interpolation_len: c_int,
12083 );
12084 pub fn atg_quantile_scalar(
12085 out__: *mut *mut C_tensor,
12086 self_: *mut C_tensor,
12087 q_: f64,
12088 dim_v: i64,
12089 dim_null: i8,
12090 keepdim_: c_int,
12091 interpolation_ptr: *const u8,
12092 interpolation_len: c_int,
12093 );
12094 pub fn atg_quantile_scalar_out(
12095 out__: *mut *mut C_tensor,
12096 out_: *mut C_tensor,
12097 self_: *mut C_tensor,
12098 q_: f64,
12099 dim_v: i64,
12100 dim_null: i8,
12101 keepdim_: c_int,
12102 interpolation_ptr: *const u8,
12103 interpolation_len: c_int,
12104 );
12105 pub fn atg_quantize_per_channel(
12106 out__: *mut *mut C_tensor,
12107 self_: *mut C_tensor,
12108 scales_: *mut C_tensor,
12109 zero_points_: *mut C_tensor,
12110 axis_: i64,
12111 dtype_: c_int,
12112 );
12113 pub fn atg_quantize_per_channel_out(
12114 out__: *mut *mut C_tensor,
12115 out_: *mut C_tensor,
12116 self_: *mut C_tensor,
12117 scales_: *mut C_tensor,
12118 zero_points_: *mut C_tensor,
12119 axis_: i64,
12120 dtype_: c_int,
12121 );
12122 pub fn atg_quantize_per_tensor(
12123 out__: *mut *mut C_tensor,
12124 self_: *mut C_tensor,
12125 scale_: f64,
12126 zero_point_: i64,
12127 dtype_: c_int,
12128 );
12129 pub fn atg_quantize_per_tensor_dynamic(
12130 out__: *mut *mut C_tensor,
12131 self_: *mut C_tensor,
12132 dtype_: c_int,
12133 reduce_range_: c_int,
12134 );
12135 pub fn atg_quantize_per_tensor_dynamic_out(
12136 out__: *mut *mut C_tensor,
12137 out_: *mut C_tensor,
12138 self_: *mut C_tensor,
12139 dtype_: c_int,
12140 reduce_range_: c_int,
12141 );
12142 pub fn atg_quantize_per_tensor_out(
12143 out__: *mut *mut C_tensor,
12144 out_: *mut C_tensor,
12145 self_: *mut C_tensor,
12146 scale_: f64,
12147 zero_point_: i64,
12148 dtype_: c_int,
12149 );
12150 pub fn atg_quantize_per_tensor_tensor_qparams(
12151 out__: *mut *mut C_tensor,
12152 self_: *mut C_tensor,
12153 scale_: *mut C_tensor,
12154 zero_point_: *mut C_tensor,
12155 dtype_: c_int,
12156 );
12157 pub fn atg_quantize_per_tensor_tensor_qparams_out(
12158 out__: *mut *mut C_tensor,
12159 out_: *mut C_tensor,
12160 self_: *mut C_tensor,
12161 scale_: *mut C_tensor,
12162 zero_point_: *mut C_tensor,
12163 dtype_: c_int,
12164 );
12165 pub fn atg_quantize_per_tensor_tensors(
12166 tensors_data: *const *mut C_tensor,
12167 tensors_len: c_int,
12168 scales_: *mut C_tensor,
12169 zero_points_: *mut C_tensor,
12170 dtype_: c_int,
12171 ) -> *mut *mut C_tensor;
12172 pub fn atg_quantize_per_tensor_tensors_out(
12173 out_data: *const *mut C_tensor,
12174 out_len: c_int,
12175 tensors_data: *const *mut C_tensor,
12176 tensors_len: c_int,
12177 scales_: *mut C_tensor,
12178 zero_points_: *mut C_tensor,
12179 dtype_: c_int,
12180 );
12181 pub fn atg_quantized_batch_norm(
12182 out__: *mut *mut C_tensor,
12183 input_: *mut C_tensor,
12184 weight_: *mut C_tensor,
12185 bias_: *mut C_tensor,
12186 mean_: *mut C_tensor,
12187 var_: *mut C_tensor,
12188 eps_: f64,
12189 output_scale_: f64,
12190 output_zero_point_: i64,
12191 );
12192 pub fn atg_quantized_batch_norm_out(
12193 out__: *mut *mut C_tensor,
12194 out_: *mut C_tensor,
12195 input_: *mut C_tensor,
12196 weight_: *mut C_tensor,
12197 bias_: *mut C_tensor,
12198 mean_: *mut C_tensor,
12199 var_: *mut C_tensor,
12200 eps_: f64,
12201 output_scale_: f64,
12202 output_zero_point_: i64,
12203 );
12204 pub fn atg_quantized_gru_cell(
12205 out__: *mut *mut C_tensor,
12206 input_: *mut C_tensor,
12207 hx_: *mut C_tensor,
12208 w_ih_: *mut C_tensor,
12209 w_hh_: *mut C_tensor,
12210 b_ih_: *mut C_tensor,
12211 b_hh_: *mut C_tensor,
12212 packed_ih_: *mut C_tensor,
12213 packed_hh_: *mut C_tensor,
12214 col_offsets_ih_: *mut C_tensor,
12215 col_offsets_hh_: *mut C_tensor,
12216 scale_ih_: *mut C_scalar,
12217 scale_hh_: *mut C_scalar,
12218 zero_point_ih_: *mut C_scalar,
12219 zero_point_hh_: *mut C_scalar,
12220 );
12221 pub fn atg_quantized_lstm_cell(
12222 out__: *mut *mut C_tensor,
12223 input_: *mut C_tensor,
12224 hx_data: *const *mut C_tensor,
12225 hx_len: c_int,
12226 w_ih_: *mut C_tensor,
12227 w_hh_: *mut C_tensor,
12228 b_ih_: *mut C_tensor,
12229 b_hh_: *mut C_tensor,
12230 packed_ih_: *mut C_tensor,
12231 packed_hh_: *mut C_tensor,
12232 col_offsets_ih_: *mut C_tensor,
12233 col_offsets_hh_: *mut C_tensor,
12234 scale_ih_: *mut C_scalar,
12235 scale_hh_: *mut C_scalar,
12236 zero_point_ih_: *mut C_scalar,
12237 zero_point_hh_: *mut C_scalar,
12238 );
12239 pub fn atg_quantized_max_pool1d(
12240 out__: *mut *mut C_tensor,
12241 self_: *mut C_tensor,
12242 kernel_size_data: *const i64,
12243 kernel_size_len: c_int,
12244 stride_data: *const i64,
12245 stride_len: c_int,
12246 padding_data: *const i64,
12247 padding_len: c_int,
12248 dilation_data: *const i64,
12249 dilation_len: c_int,
12250 ceil_mode_: c_int,
12251 );
12252 pub fn atg_quantized_max_pool1d_out(
12253 out__: *mut *mut C_tensor,
12254 out_: *mut C_tensor,
12255 self_: *mut C_tensor,
12256 kernel_size_data: *const i64,
12257 kernel_size_len: c_int,
12258 stride_data: *const i64,
12259 stride_len: c_int,
12260 padding_data: *const i64,
12261 padding_len: c_int,
12262 dilation_data: *const i64,
12263 dilation_len: c_int,
12264 ceil_mode_: c_int,
12265 );
12266 pub fn atg_quantized_max_pool2d(
12267 out__: *mut *mut C_tensor,
12268 self_: *mut C_tensor,
12269 kernel_size_data: *const i64,
12270 kernel_size_len: c_int,
12271 stride_data: *const i64,
12272 stride_len: c_int,
12273 padding_data: *const i64,
12274 padding_len: c_int,
12275 dilation_data: *const i64,
12276 dilation_len: c_int,
12277 ceil_mode_: c_int,
12278 );
12279 pub fn atg_quantized_max_pool2d_out(
12280 out__: *mut *mut C_tensor,
12281 out_: *mut C_tensor,
12282 self_: *mut C_tensor,
12283 kernel_size_data: *const i64,
12284 kernel_size_len: c_int,
12285 stride_data: *const i64,
12286 stride_len: c_int,
12287 padding_data: *const i64,
12288 padding_len: c_int,
12289 dilation_data: *const i64,
12290 dilation_len: c_int,
12291 ceil_mode_: c_int,
12292 );
12293 pub fn atg_quantized_max_pool3d(
12294 out__: *mut *mut C_tensor,
12295 self_: *mut C_tensor,
12296 kernel_size_data: *const i64,
12297 kernel_size_len: c_int,
12298 stride_data: *const i64,
12299 stride_len: c_int,
12300 padding_data: *const i64,
12301 padding_len: c_int,
12302 dilation_data: *const i64,
12303 dilation_len: c_int,
12304 ceil_mode_: c_int,
12305 );
12306 pub fn atg_quantized_max_pool3d_out(
12307 out__: *mut *mut C_tensor,
12308 out_: *mut C_tensor,
12309 self_: *mut C_tensor,
12310 kernel_size_data: *const i64,
12311 kernel_size_len: c_int,
12312 stride_data: *const i64,
12313 stride_len: c_int,
12314 padding_data: *const i64,
12315 padding_len: c_int,
12316 dilation_data: *const i64,
12317 dilation_len: c_int,
12318 ceil_mode_: c_int,
12319 );
12320 pub fn atg_quantized_rnn_relu_cell(
12321 out__: *mut *mut C_tensor,
12322 input_: *mut C_tensor,
12323 hx_: *mut C_tensor,
12324 w_ih_: *mut C_tensor,
12325 w_hh_: *mut C_tensor,
12326 b_ih_: *mut C_tensor,
12327 b_hh_: *mut C_tensor,
12328 packed_ih_: *mut C_tensor,
12329 packed_hh_: *mut C_tensor,
12330 col_offsets_ih_: *mut C_tensor,
12331 col_offsets_hh_: *mut C_tensor,
12332 scale_ih_: *mut C_scalar,
12333 scale_hh_: *mut C_scalar,
12334 zero_point_ih_: *mut C_scalar,
12335 zero_point_hh_: *mut C_scalar,
12336 );
12337 pub fn atg_quantized_rnn_tanh_cell(
12338 out__: *mut *mut C_tensor,
12339 input_: *mut C_tensor,
12340 hx_: *mut C_tensor,
12341 w_ih_: *mut C_tensor,
12342 w_hh_: *mut C_tensor,
12343 b_ih_: *mut C_tensor,
12344 b_hh_: *mut C_tensor,
12345 packed_ih_: *mut C_tensor,
12346 packed_hh_: *mut C_tensor,
12347 col_offsets_ih_: *mut C_tensor,
12348 col_offsets_hh_: *mut C_tensor,
12349 scale_ih_: *mut C_scalar,
12350 scale_hh_: *mut C_scalar,
12351 zero_point_ih_: *mut C_scalar,
12352 zero_point_hh_: *mut C_scalar,
12353 );
12354 pub fn atg_rad2deg(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12355 pub fn atg_rad2deg_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12356 pub fn atg_rad2deg_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
12357 pub fn atg_rand(
12358 out__: *mut *mut C_tensor,
12359 size_data: *const i64,
12360 size_len: c_int,
12361 options_kind: c_int,
12362 options_device: c_int,
12363 );
12364 pub fn atg_rand_like(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12365 pub fn atg_rand_like_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
12366 pub fn atg_rand_out(
12367 out__: *mut *mut C_tensor,
12368 out_: *mut C_tensor,
12369 size_data: *const i64,
12370 size_len: c_int,
12371 );
12372 pub fn atg_randint(
12373 out__: *mut *mut C_tensor,
12374 high_: i64,
12375 size_data: *const i64,
12376 size_len: c_int,
12377 options_kind: c_int,
12378 options_device: c_int,
12379 );
12380 pub fn atg_randint_like(out__: *mut *mut C_tensor, self_: *mut C_tensor, high_: i64);
12381 pub fn atg_randint_like_low_dtype(
12382 out__: *mut *mut C_tensor,
12383 self_: *mut C_tensor,
12384 low_: i64,
12385 high_: i64,
12386 );
12387 pub fn atg_randint_like_low_dtype_out(
12388 out__: *mut *mut C_tensor,
12389 out_: *mut C_tensor,
12390 self_: *mut C_tensor,
12391 low_: i64,
12392 high_: i64,
12393 );
12394 pub fn atg_randint_like_out(
12395 out__: *mut *mut C_tensor,
12396 out_: *mut C_tensor,
12397 self_: *mut C_tensor,
12398 high_: i64,
12399 );
12400 pub fn atg_randint_like_tensor(
12401 out__: *mut *mut C_tensor,
12402 self_: *mut C_tensor,
12403 high_: *mut C_tensor,
12404 );
12405 pub fn atg_randint_like_tensor_out(
12406 out__: *mut *mut C_tensor,
12407 out_: *mut C_tensor,
12408 self_: *mut C_tensor,
12409 high_: *mut C_tensor,
12410 );
12411 pub fn atg_randint_low(
12412 out__: *mut *mut C_tensor,
12413 low_: i64,
12414 high_: i64,
12415 size_data: *const i64,
12416 size_len: c_int,
12417 options_kind: c_int,
12418 options_device: c_int,
12419 );
12420 pub fn atg_randint_low_out(
12421 out__: *mut *mut C_tensor,
12422 out_: *mut C_tensor,
12423 low_: i64,
12424 high_: i64,
12425 size_data: *const i64,
12426 size_len: c_int,
12427 );
12428 pub fn atg_randint_out(
12429 out__: *mut *mut C_tensor,
12430 out_: *mut C_tensor,
12431 high_: i64,
12432 size_data: *const i64,
12433 size_len: c_int,
12434 );
12435 pub fn atg_randn(
12436 out__: *mut *mut C_tensor,
12437 size_data: *const i64,
12438 size_len: c_int,
12439 options_kind: c_int,
12440 options_device: c_int,
12441 );
12442 pub fn atg_randn_like(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12443 pub fn atg_randn_like_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
12444 pub fn atg_randn_out(
12445 out__: *mut *mut C_tensor,
12446 out_: *mut C_tensor,
12447 size_data: *const i64,
12448 size_len: c_int,
12449 );
12450 pub fn atg_random(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12451 pub fn atg_random_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12452 pub fn atg_random_from(
12453 out__: *mut *mut C_tensor,
12454 self_: *mut C_tensor,
12455 from_: i64,
12456 to_v: i64,
12457 to_null: i8,
12458 );
12459 pub fn atg_random_from_(
12460 out__: *mut *mut C_tensor,
12461 self_: *mut C_tensor,
12462 from_: i64,
12463 to_v: i64,
12464 to_null: i8,
12465 );
12466 pub fn atg_random_from_out(
12467 out__: *mut *mut C_tensor,
12468 out_: *mut C_tensor,
12469 self_: *mut C_tensor,
12470 from_: i64,
12471 to_v: i64,
12472 to_null: i8,
12473 );
12474 pub fn atg_random_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
12475 pub fn atg_random_to(out__: *mut *mut C_tensor, self_: *mut C_tensor, to_: i64);
12476 pub fn atg_random_to_(out__: *mut *mut C_tensor, self_: *mut C_tensor, to_: i64);
12477 pub fn atg_random_to_out(
12478 out__: *mut *mut C_tensor,
12479 out_: *mut C_tensor,
12480 self_: *mut C_tensor,
12481 to_: i64,
12482 );
12483 pub fn atg_randperm(
12484 out__: *mut *mut C_tensor,
12485 n_: i64,
12486 options_kind: c_int,
12487 options_device: c_int,
12488 );
12489 pub fn atg_randperm_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, n_: i64);
12490 pub fn atg_range(
12491 out__: *mut *mut C_tensor,
12492 start_: *mut C_scalar,
12493 end_: *mut C_scalar,
12494 options_kind: c_int,
12495 options_device: c_int,
12496 );
12497 pub fn atg_range_out(
12498 out__: *mut *mut C_tensor,
12499 out_: *mut C_tensor,
12500 start_: *mut C_scalar,
12501 end_: *mut C_scalar,
12502 );
12503 pub fn atg_range_out_(
12504 out__: *mut *mut C_tensor,
12505 out_: *mut C_tensor,
12506 start_: *mut C_scalar,
12507 end_: *mut C_scalar,
12508 );
12509 pub fn atg_range_step(
12510 out__: *mut *mut C_tensor,
12511 start_: *mut C_scalar,
12512 end_: *mut C_scalar,
12513 options_kind: c_int,
12514 options_device: c_int,
12515 );
12516 pub fn atg_ravel(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12517 pub fn atg_real(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12518 pub fn atg_reciprocal(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12519 pub fn atg_reciprocal_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12520 pub fn atg_reciprocal_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
12521 pub fn atg_reflection_pad1d(
12522 out__: *mut *mut C_tensor,
12523 self_: *mut C_tensor,
12524 padding_data: *const i64,
12525 padding_len: c_int,
12526 );
12527 pub fn atg_reflection_pad1d_backward(
12528 out__: *mut *mut C_tensor,
12529 grad_output_: *mut C_tensor,
12530 self_: *mut C_tensor,
12531 padding_data: *const i64,
12532 padding_len: c_int,
12533 );
12534 pub fn atg_reflection_pad1d_backward_grad_input(
12535 out__: *mut *mut C_tensor,
12536 grad_input_: *mut C_tensor,
12537 grad_output_: *mut C_tensor,
12538 self_: *mut C_tensor,
12539 padding_data: *const i64,
12540 padding_len: c_int,
12541 );
12542 pub fn atg_reflection_pad1d_out(
12543 out__: *mut *mut C_tensor,
12544 out_: *mut C_tensor,
12545 self_: *mut C_tensor,
12546 padding_data: *const i64,
12547 padding_len: c_int,
12548 );
12549 pub fn atg_reflection_pad2d(
12550 out__: *mut *mut C_tensor,
12551 self_: *mut C_tensor,
12552 padding_data: *const i64,
12553 padding_len: c_int,
12554 );
12555 pub fn atg_reflection_pad2d_backward(
12556 out__: *mut *mut C_tensor,
12557 grad_output_: *mut C_tensor,
12558 self_: *mut C_tensor,
12559 padding_data: *const i64,
12560 padding_len: c_int,
12561 );
12562 pub fn atg_reflection_pad2d_backward_grad_input(
12563 out__: *mut *mut C_tensor,
12564 grad_input_: *mut C_tensor,
12565 grad_output_: *mut C_tensor,
12566 self_: *mut C_tensor,
12567 padding_data: *const i64,
12568 padding_len: c_int,
12569 );
12570 pub fn atg_reflection_pad2d_out(
12571 out__: *mut *mut C_tensor,
12572 out_: *mut C_tensor,
12573 self_: *mut C_tensor,
12574 padding_data: *const i64,
12575 padding_len: c_int,
12576 );
12577 pub fn atg_reflection_pad3d(
12578 out__: *mut *mut C_tensor,
12579 self_: *mut C_tensor,
12580 padding_data: *const i64,
12581 padding_len: c_int,
12582 );
12583 pub fn atg_reflection_pad3d_backward(
12584 out__: *mut *mut C_tensor,
12585 grad_output_: *mut C_tensor,
12586 self_: *mut C_tensor,
12587 padding_data: *const i64,
12588 padding_len: c_int,
12589 );
12590 pub fn atg_reflection_pad3d_backward_grad_input(
12591 out__: *mut *mut C_tensor,
12592 grad_input_: *mut C_tensor,
12593 grad_output_: *mut C_tensor,
12594 self_: *mut C_tensor,
12595 padding_data: *const i64,
12596 padding_len: c_int,
12597 );
12598 pub fn atg_reflection_pad3d_out(
12599 out__: *mut *mut C_tensor,
12600 out_: *mut C_tensor,
12601 self_: *mut C_tensor,
12602 padding_data: *const i64,
12603 padding_len: c_int,
12604 );
12605 pub fn atg_relu(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12606 pub fn atg_relu6(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12607 pub fn atg_relu6_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12608 pub fn atg_relu_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12609 pub fn atg_relu_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
12610 pub fn atg_remainder(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
12611 pub fn atg_remainder_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
12612 pub fn atg_remainder_scalar_out(
12613 out__: *mut *mut C_tensor,
12614 out_: *mut C_tensor,
12615 self_: *mut C_tensor,
12616 other_: *mut C_scalar,
12617 );
12618 pub fn atg_remainder_scalar_tensor(
12619 out__: *mut *mut C_tensor,
12620 self_scalar_: *mut C_scalar,
12621 other_: *mut C_tensor,
12622 );
12623 pub fn atg_remainder_scalar_tensor_out(
12624 out__: *mut *mut C_tensor,
12625 out_: *mut C_tensor,
12626 self_scalar_: *mut C_scalar,
12627 other_: *mut C_tensor,
12628 );
12629 pub fn atg_remainder_tensor(
12630 out__: *mut *mut C_tensor,
12631 self_: *mut C_tensor,
12632 other_: *mut C_tensor,
12633 );
12634 pub fn atg_remainder_tensor_(
12635 out__: *mut *mut C_tensor,
12636 self_: *mut C_tensor,
12637 other_: *mut C_tensor,
12638 );
12639 pub fn atg_remainder_tensor_out(
12640 out__: *mut *mut C_tensor,
12641 out_: *mut C_tensor,
12642 self_: *mut C_tensor,
12643 other_: *mut C_tensor,
12644 );
12645 pub fn atg_renorm(
12646 out__: *mut *mut C_tensor,
12647 self_: *mut C_tensor,
12648 p_: *mut C_scalar,
12649 dim_: i64,
12650 maxnorm_: *mut C_scalar,
12651 );
12652 pub fn atg_renorm_(
12653 out__: *mut *mut C_tensor,
12654 self_: *mut C_tensor,
12655 p_: *mut C_scalar,
12656 dim_: i64,
12657 maxnorm_: *mut C_scalar,
12658 );
12659 pub fn atg_renorm_out(
12660 out__: *mut *mut C_tensor,
12661 out_: *mut C_tensor,
12662 self_: *mut C_tensor,
12663 p_: *mut C_scalar,
12664 dim_: i64,
12665 maxnorm_: *mut C_scalar,
12666 );
12667 pub fn atg_repeat(
12668 out__: *mut *mut C_tensor,
12669 self_: *mut C_tensor,
12670 repeats_data: *const i64,
12671 repeats_len: c_int,
12672 );
12673 pub fn atg_repeat_interleave(
12674 out__: *mut *mut C_tensor,
12675 repeats_: *mut C_tensor,
12676 output_size_v: i64,
12677 output_size_null: i8,
12678 );
12679 pub fn atg_repeat_interleave_self_int(
12680 out__: *mut *mut C_tensor,
12681 self_: *mut C_tensor,
12682 repeats_: i64,
12683 dim_v: i64,
12684 dim_null: i8,
12685 output_size_v: i64,
12686 output_size_null: i8,
12687 );
12688 pub fn atg_repeat_interleave_self_tensor(
12689 out__: *mut *mut C_tensor,
12690 self_: *mut C_tensor,
12691 repeats_: *mut C_tensor,
12692 dim_v: i64,
12693 dim_null: i8,
12694 output_size_v: i64,
12695 output_size_null: i8,
12696 );
12697 pub fn atg_repeat_interleave_tensor_out(
12698 out__: *mut *mut C_tensor,
12699 out_: *mut C_tensor,
12700 repeats_: *mut C_tensor,
12701 output_size_v: i64,
12702 output_size_null: i8,
12703 );
12704 pub fn atg_repeat_out(
12705 out__: *mut *mut C_tensor,
12706 out_: *mut C_tensor,
12707 self_: *mut C_tensor,
12708 repeats_data: *const i64,
12709 repeats_len: c_int,
12710 );
12711 pub fn atg_replication_pad1d(
12712 out__: *mut *mut C_tensor,
12713 self_: *mut C_tensor,
12714 padding_data: *const i64,
12715 padding_len: c_int,
12716 );
12717 pub fn atg_replication_pad1d_backward(
12718 out__: *mut *mut C_tensor,
12719 grad_output_: *mut C_tensor,
12720 self_: *mut C_tensor,
12721 padding_data: *const i64,
12722 padding_len: c_int,
12723 );
12724 pub fn atg_replication_pad1d_backward_grad_input(
12725 out__: *mut *mut C_tensor,
12726 grad_input_: *mut C_tensor,
12727 grad_output_: *mut C_tensor,
12728 self_: *mut C_tensor,
12729 padding_data: *const i64,
12730 padding_len: c_int,
12731 );
12732 pub fn atg_replication_pad1d_out(
12733 out__: *mut *mut C_tensor,
12734 out_: *mut C_tensor,
12735 self_: *mut C_tensor,
12736 padding_data: *const i64,
12737 padding_len: c_int,
12738 );
12739 pub fn atg_replication_pad2d(
12740 out__: *mut *mut C_tensor,
12741 self_: *mut C_tensor,
12742 padding_data: *const i64,
12743 padding_len: c_int,
12744 );
12745 pub fn atg_replication_pad2d_backward(
12746 out__: *mut *mut C_tensor,
12747 grad_output_: *mut C_tensor,
12748 self_: *mut C_tensor,
12749 padding_data: *const i64,
12750 padding_len: c_int,
12751 );
12752 pub fn atg_replication_pad2d_backward_grad_input(
12753 out__: *mut *mut C_tensor,
12754 grad_input_: *mut C_tensor,
12755 grad_output_: *mut C_tensor,
12756 self_: *mut C_tensor,
12757 padding_data: *const i64,
12758 padding_len: c_int,
12759 );
12760 pub fn atg_replication_pad2d_out(
12761 out__: *mut *mut C_tensor,
12762 out_: *mut C_tensor,
12763 self_: *mut C_tensor,
12764 padding_data: *const i64,
12765 padding_len: c_int,
12766 );
12767 pub fn atg_replication_pad3d(
12768 out__: *mut *mut C_tensor,
12769 self_: *mut C_tensor,
12770 padding_data: *const i64,
12771 padding_len: c_int,
12772 );
12773 pub fn atg_replication_pad3d_backward(
12774 out__: *mut *mut C_tensor,
12775 grad_output_: *mut C_tensor,
12776 self_: *mut C_tensor,
12777 padding_data: *const i64,
12778 padding_len: c_int,
12779 );
12780 pub fn atg_replication_pad3d_backward_grad_input(
12781 out__: *mut *mut C_tensor,
12782 grad_input_: *mut C_tensor,
12783 grad_output_: *mut C_tensor,
12784 self_: *mut C_tensor,
12785 padding_data: *const i64,
12786 padding_len: c_int,
12787 );
12788 pub fn atg_replication_pad3d_out(
12789 out__: *mut *mut C_tensor,
12790 out_: *mut C_tensor,
12791 self_: *mut C_tensor,
12792 padding_data: *const i64,
12793 padding_len: c_int,
12794 );
12795 pub fn atg_requires_grad_(
12796 out__: *mut *mut C_tensor,
12797 self_: *mut C_tensor,
12798 requires_grad_: c_int,
12799 );
12800 pub fn atg_reshape(
12801 out__: *mut *mut C_tensor,
12802 self_: *mut C_tensor,
12803 shape_data: *const i64,
12804 shape_len: c_int,
12805 );
12806 pub fn atg_reshape_as(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
12807 pub fn atg_resize(
12808 out__: *mut *mut C_tensor,
12809 self_: *mut C_tensor,
12810 size_data: *const i64,
12811 size_len: c_int,
12812 );
12813 pub fn atg_resize_(
12814 out__: *mut *mut C_tensor,
12815 self_: *mut C_tensor,
12816 size_data: *const i64,
12817 size_len: c_int,
12818 );
12819 pub fn atg_resize_as(
12820 out__: *mut *mut C_tensor,
12821 self_: *mut C_tensor,
12822 the_template_: *mut C_tensor,
12823 );
12824 pub fn atg_resize_as_(
12825 out__: *mut *mut C_tensor,
12826 self_: *mut C_tensor,
12827 the_template_: *mut C_tensor,
12828 );
12829 pub fn atg_resize_as_out(
12830 out__: *mut *mut C_tensor,
12831 out_: *mut C_tensor,
12832 self_: *mut C_tensor,
12833 the_template_: *mut C_tensor,
12834 );
12835 pub fn atg_resize_as_sparse(
12836 out__: *mut *mut C_tensor,
12837 self_: *mut C_tensor,
12838 the_template_: *mut C_tensor,
12839 );
12840 pub fn atg_resize_as_sparse_(
12841 out__: *mut *mut C_tensor,
12842 self_: *mut C_tensor,
12843 the_template_: *mut C_tensor,
12844 );
12845 pub fn atg_resize_as_sparse_out(
12846 out__: *mut *mut C_tensor,
12847 out_: *mut C_tensor,
12848 self_: *mut C_tensor,
12849 the_template_: *mut C_tensor,
12850 );
12851 pub fn atg_resize_out(
12852 out__: *mut *mut C_tensor,
12853 out_: *mut C_tensor,
12854 self_: *mut C_tensor,
12855 size_data: *const i64,
12856 size_len: c_int,
12857 );
12858 pub fn atg_resolve_conj(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12859 pub fn atg_resolve_neg(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12860 pub fn atg_retains_grad(self_: *mut C_tensor) -> c_int;
12861 pub fn atg_rms_norm(
12862 out__: *mut *mut C_tensor,
12863 input_: *mut C_tensor,
12864 normalized_shape_data: *const i64,
12865 normalized_shape_len: c_int,
12866 weight_: *mut C_tensor,
12867 eps_v: f64,
12868 eps_null: i8,
12869 );
12870 pub fn atg_rnn_relu(
12871 out__: *mut *mut C_tensor,
12872 input_: *mut C_tensor,
12873 hx_: *mut C_tensor,
12874 params_data: *const *mut C_tensor,
12875 params_len: c_int,
12876 has_biases_: c_int,
12877 num_layers_: i64,
12878 dropout_: f64,
12879 train_: c_int,
12880 bidirectional_: c_int,
12881 batch_first_: c_int,
12882 );
12883 pub fn atg_rnn_relu_cell(
12884 out__: *mut *mut C_tensor,
12885 input_: *mut C_tensor,
12886 hx_: *mut C_tensor,
12887 w_ih_: *mut C_tensor,
12888 w_hh_: *mut C_tensor,
12889 b_ih_: *mut C_tensor,
12890 b_hh_: *mut C_tensor,
12891 );
12892 pub fn atg_rnn_relu_data(
12893 out__: *mut *mut C_tensor,
12894 data_: *mut C_tensor,
12895 batch_sizes_: *mut C_tensor,
12896 hx_: *mut C_tensor,
12897 params_data: *const *mut C_tensor,
12898 params_len: c_int,
12899 has_biases_: c_int,
12900 num_layers_: i64,
12901 dropout_: f64,
12902 train_: c_int,
12903 bidirectional_: c_int,
12904 );
12905 pub fn atg_rnn_tanh(
12906 out__: *mut *mut C_tensor,
12907 input_: *mut C_tensor,
12908 hx_: *mut C_tensor,
12909 params_data: *const *mut C_tensor,
12910 params_len: c_int,
12911 has_biases_: c_int,
12912 num_layers_: i64,
12913 dropout_: f64,
12914 train_: c_int,
12915 bidirectional_: c_int,
12916 batch_first_: c_int,
12917 );
12918 pub fn atg_rnn_tanh_cell(
12919 out__: *mut *mut C_tensor,
12920 input_: *mut C_tensor,
12921 hx_: *mut C_tensor,
12922 w_ih_: *mut C_tensor,
12923 w_hh_: *mut C_tensor,
12924 b_ih_: *mut C_tensor,
12925 b_hh_: *mut C_tensor,
12926 );
12927 pub fn atg_rnn_tanh_data(
12928 out__: *mut *mut C_tensor,
12929 data_: *mut C_tensor,
12930 batch_sizes_: *mut C_tensor,
12931 hx_: *mut C_tensor,
12932 params_data: *const *mut C_tensor,
12933 params_len: c_int,
12934 has_biases_: c_int,
12935 num_layers_: i64,
12936 dropout_: f64,
12937 train_: c_int,
12938 bidirectional_: c_int,
12939 );
12940 pub fn atg_roll(
12941 out__: *mut *mut C_tensor,
12942 self_: *mut C_tensor,
12943 shifts_data: *const i64,
12944 shifts_len: c_int,
12945 dims_data: *const i64,
12946 dims_len: c_int,
12947 );
12948 pub fn atg_roll_out(
12949 out__: *mut *mut C_tensor,
12950 out_: *mut C_tensor,
12951 self_: *mut C_tensor,
12952 shifts_data: *const i64,
12953 shifts_len: c_int,
12954 dims_data: *const i64,
12955 dims_len: c_int,
12956 );
12957 pub fn atg_rot90(
12958 out__: *mut *mut C_tensor,
12959 self_: *mut C_tensor,
12960 k_: i64,
12961 dims_data: *const i64,
12962 dims_len: c_int,
12963 );
12964 pub fn atg_rot90_out(
12965 out__: *mut *mut C_tensor,
12966 out_: *mut C_tensor,
12967 self_: *mut C_tensor,
12968 k_: i64,
12969 dims_data: *const i64,
12970 dims_len: c_int,
12971 );
12972 pub fn atg_round(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12973 pub fn atg_round_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12974 pub fn atg_round_decimals(out__: *mut *mut C_tensor, self_: *mut C_tensor, decimals_: i64);
12975 pub fn atg_round_decimals_(out__: *mut *mut C_tensor, self_: *mut C_tensor, decimals_: i64);
12976 pub fn atg_round_decimals_out(
12977 out__: *mut *mut C_tensor,
12978 out_: *mut C_tensor,
12979 self_: *mut C_tensor,
12980 decimals_: i64,
12981 );
12982 pub fn atg_round_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
12983 pub fn atg_row_indices(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12984 pub fn atg_row_indices_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
12985 pub fn atg_row_indices_copy_out(
12986 out__: *mut *mut C_tensor,
12987 out_: *mut C_tensor,
12988 self_: *mut C_tensor,
12989 );
12990 pub fn atg_row_stack(
12991 out__: *mut *mut C_tensor,
12992 tensors_data: *const *mut C_tensor,
12993 tensors_len: c_int,
12994 );
12995 pub fn atg_row_stack_out(
12996 out__: *mut *mut C_tensor,
12997 out_: *mut C_tensor,
12998 tensors_data: *const *mut C_tensor,
12999 tensors_len: c_int,
13000 );
13001 pub fn atg_rrelu(out__: *mut *mut C_tensor, self_: *mut C_tensor, training_: c_int);
13002 pub fn atg_rrelu_(out__: *mut *mut C_tensor, self_: *mut C_tensor, training_: c_int);
13003 pub fn atg_rrelu_with_noise(
13004 out__: *mut *mut C_tensor,
13005 self_: *mut C_tensor,
13006 noise_: *mut C_tensor,
13007 training_: c_int,
13008 );
13009 pub fn atg_rrelu_with_noise_(
13010 out__: *mut *mut C_tensor,
13011 self_: *mut C_tensor,
13012 noise_: *mut C_tensor,
13013 training_: c_int,
13014 );
13015 pub fn atg_rrelu_with_noise_backward(
13016 out__: *mut *mut C_tensor,
13017 grad_output_: *mut C_tensor,
13018 self_: *mut C_tensor,
13019 noise_: *mut C_tensor,
13020 lower_: *mut C_scalar,
13021 upper_: *mut C_scalar,
13022 training_: c_int,
13023 self_is_result_: c_int,
13024 );
13025 pub fn atg_rrelu_with_noise_backward_out(
13026 out__: *mut *mut C_tensor,
13027 out_: *mut C_tensor,
13028 grad_output_: *mut C_tensor,
13029 self_: *mut C_tensor,
13030 noise_: *mut C_tensor,
13031 lower_: *mut C_scalar,
13032 upper_: *mut C_scalar,
13033 training_: c_int,
13034 self_is_result_: c_int,
13035 );
13036 pub fn atg_rrelu_with_noise_functional(
13037 out__: *mut *mut C_tensor,
13038 self_: *mut C_tensor,
13039 noise_: *mut C_tensor,
13040 training_: c_int,
13041 );
13042 pub fn atg_rrelu_with_noise_out(
13043 out__: *mut *mut C_tensor,
13044 out_: *mut C_tensor,
13045 self_: *mut C_tensor,
13046 noise_: *mut C_tensor,
13047 training_: c_int,
13048 );
13049 pub fn atg_rsqrt(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13050 pub fn atg_rsqrt_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13051 pub fn atg_rsqrt_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13052 pub fn atg_rsub(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
13053 pub fn atg_rsub_scalar(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
13054 pub fn atg_rsub_scalar_out(
13055 out__: *mut *mut C_tensor,
13056 out_: *mut C_tensor,
13057 self_: *mut C_tensor,
13058 other_: *mut C_scalar,
13059 );
13060 pub fn atg_rsub_tensor_out(
13061 out__: *mut *mut C_tensor,
13062 out_: *mut C_tensor,
13063 self_: *mut C_tensor,
13064 other_: *mut C_tensor,
13065 );
13066 pub fn atg_scalar_tensor(
13067 out__: *mut *mut C_tensor,
13068 s_: *mut C_scalar,
13069 options_kind: c_int,
13070 options_device: c_int,
13071 );
13072 pub fn atg_scalar_tensor_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, s_: *mut C_scalar);
13073 pub fn atg_scaled_dot_product_attention(
13074 out__: *mut *mut C_tensor,
13075 query_: *mut C_tensor,
13076 key_: *mut C_tensor,
13077 value_: *mut C_tensor,
13078 attn_mask_: *mut C_tensor,
13079 dropout_p_: f64,
13080 is_causal_: c_int,
13081 scale_v: f64,
13082 scale_null: i8,
13083 enable_gqa_: c_int,
13084 );
13085 pub fn atg_scatter(
13086 out__: *mut *mut C_tensor,
13087 self_: *mut C_tensor,
13088 dim_: i64,
13089 index_: *mut C_tensor,
13090 src_: *mut C_tensor,
13091 );
13092 pub fn atg_scatter_(
13093 out__: *mut *mut C_tensor,
13094 self_: *mut C_tensor,
13095 dim_: i64,
13096 index_: *mut C_tensor,
13097 src_: *mut C_tensor,
13098 );
13099 pub fn atg_scatter_add(
13100 out__: *mut *mut C_tensor,
13101 self_: *mut C_tensor,
13102 dim_: i64,
13103 index_: *mut C_tensor,
13104 src_: *mut C_tensor,
13105 );
13106 pub fn atg_scatter_add_(
13107 out__: *mut *mut C_tensor,
13108 self_: *mut C_tensor,
13109 dim_: i64,
13110 index_: *mut C_tensor,
13111 src_: *mut C_tensor,
13112 );
13113 pub fn atg_scatter_add_out(
13114 out__: *mut *mut C_tensor,
13115 out_: *mut C_tensor,
13116 self_: *mut C_tensor,
13117 dim_: i64,
13118 index_: *mut C_tensor,
13119 src_: *mut C_tensor,
13120 );
13121 pub fn atg_scatter_reduce(
13122 out__: *mut *mut C_tensor,
13123 self_: *mut C_tensor,
13124 dim_: i64,
13125 index_: *mut C_tensor,
13126 src_: *mut C_tensor,
13127 reduce_ptr: *const u8,
13128 reduce_len: c_int,
13129 );
13130 pub fn atg_scatter_reduce_(
13131 out__: *mut *mut C_tensor,
13132 self_: *mut C_tensor,
13133 dim_: i64,
13134 index_: *mut C_tensor,
13135 src_: *mut C_tensor,
13136 reduce_ptr: *const u8,
13137 reduce_len: c_int,
13138 );
13139 pub fn atg_scatter_reduce_out(
13140 out__: *mut *mut C_tensor,
13141 out_: *mut C_tensor,
13142 self_: *mut C_tensor,
13143 dim_: i64,
13144 index_: *mut C_tensor,
13145 src_: *mut C_tensor,
13146 reduce_ptr: *const u8,
13147 reduce_len: c_int,
13148 );
13149 pub fn atg_scatter_src_out(
13150 out__: *mut *mut C_tensor,
13151 out_: *mut C_tensor,
13152 self_: *mut C_tensor,
13153 dim_: i64,
13154 index_: *mut C_tensor,
13155 src_: *mut C_tensor,
13156 );
13157 pub fn atg_scatter_value(
13158 out__: *mut *mut C_tensor,
13159 self_: *mut C_tensor,
13160 dim_: i64,
13161 index_: *mut C_tensor,
13162 value_: *mut C_scalar,
13163 );
13164 pub fn atg_scatter_value_(
13165 out__: *mut *mut C_tensor,
13166 self_: *mut C_tensor,
13167 dim_: i64,
13168 index_: *mut C_tensor,
13169 value_: *mut C_scalar,
13170 );
13171 pub fn atg_scatter_value_out(
13172 out__: *mut *mut C_tensor,
13173 out_: *mut C_tensor,
13174 self_: *mut C_tensor,
13175 dim_: i64,
13176 index_: *mut C_tensor,
13177 value_: *mut C_scalar,
13178 );
13179 pub fn atg_scatter_value_reduce(
13180 out__: *mut *mut C_tensor,
13181 self_: *mut C_tensor,
13182 dim_: i64,
13183 index_: *mut C_tensor,
13184 value_: *mut C_scalar,
13185 reduce_ptr: *const u8,
13186 reduce_len: c_int,
13187 );
13188 pub fn atg_scatter_value_reduce_(
13189 out__: *mut *mut C_tensor,
13190 self_: *mut C_tensor,
13191 dim_: i64,
13192 index_: *mut C_tensor,
13193 value_: *mut C_scalar,
13194 reduce_ptr: *const u8,
13195 reduce_len: c_int,
13196 );
13197 pub fn atg_scatter_value_reduce_out(
13198 out__: *mut *mut C_tensor,
13199 out_: *mut C_tensor,
13200 self_: *mut C_tensor,
13201 dim_: i64,
13202 index_: *mut C_tensor,
13203 value_: *mut C_scalar,
13204 reduce_ptr: *const u8,
13205 reduce_len: c_int,
13206 );
13207 pub fn atg_searchsorted(
13208 out__: *mut *mut C_tensor,
13209 sorted_sequence_: *mut C_tensor,
13210 self_: *mut C_tensor,
13211 out_int32_: c_int,
13212 right_: c_int,
13213 side_ptr: *const u8,
13214 side_len: c_int,
13215 sorter_: *mut C_tensor,
13216 );
13217 pub fn atg_searchsorted_scalar(
13218 out__: *mut *mut C_tensor,
13219 sorted_sequence_: *mut C_tensor,
13220 self_scalar_: *mut C_scalar,
13221 out_int32_: c_int,
13222 right_: c_int,
13223 side_ptr: *const u8,
13224 side_len: c_int,
13225 sorter_: *mut C_tensor,
13226 );
13227 pub fn atg_searchsorted_scalar_out(
13228 out__: *mut *mut C_tensor,
13229 out_: *mut C_tensor,
13230 sorted_sequence_: *mut C_tensor,
13231 self_scalar_: *mut C_scalar,
13232 out_int32_: c_int,
13233 right_: c_int,
13234 side_ptr: *const u8,
13235 side_len: c_int,
13236 sorter_: *mut C_tensor,
13237 );
13238 pub fn atg_searchsorted_tensor_out(
13239 out__: *mut *mut C_tensor,
13240 out_: *mut C_tensor,
13241 sorted_sequence_: *mut C_tensor,
13242 self_: *mut C_tensor,
13243 out_int32_: c_int,
13244 right_: c_int,
13245 side_ptr: *const u8,
13246 side_len: c_int,
13247 sorter_: *mut C_tensor,
13248 );
13249 pub fn atg_segment_reduce(
13250 out__: *mut *mut C_tensor,
13251 data_: *mut C_tensor,
13252 reduce_ptr: *const u8,
13253 reduce_len: c_int,
13254 lengths_: *mut C_tensor,
13255 indices_: *mut C_tensor,
13256 offsets_: *mut C_tensor,
13257 axis_: i64,
13258 unsafe_: c_int,
13259 initial_: *mut C_scalar,
13260 );
13261 pub fn atg_segment_reduce_out(
13262 out__: *mut *mut C_tensor,
13263 out_: *mut C_tensor,
13264 data_: *mut C_tensor,
13265 reduce_ptr: *const u8,
13266 reduce_len: c_int,
13267 lengths_: *mut C_tensor,
13268 indices_: *mut C_tensor,
13269 offsets_: *mut C_tensor,
13270 axis_: i64,
13271 unsafe_: c_int,
13272 initial_: *mut C_scalar,
13273 );
13274 pub fn atg_select(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, index_: i64);
13275 pub fn atg_select_backward(
13276 out__: *mut *mut C_tensor,
13277 grad_output_: *mut C_tensor,
13278 input_sizes_data: *const i64,
13279 input_sizes_len: c_int,
13280 dim_: i64,
13281 index_: i64,
13282 );
13283 pub fn atg_select_backward_out(
13284 out__: *mut *mut C_tensor,
13285 out_: *mut C_tensor,
13286 grad_output_: *mut C_tensor,
13287 input_sizes_data: *const i64,
13288 input_sizes_len: c_int,
13289 dim_: i64,
13290 index_: i64,
13291 );
13292 pub fn atg_select_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, index_: i64);
13293 pub fn atg_select_copy_int_out(
13294 out__: *mut *mut C_tensor,
13295 out_: *mut C_tensor,
13296 self_: *mut C_tensor,
13297 dim_: i64,
13298 index_: i64,
13299 );
13300 pub fn atg_select_scatter(
13301 out__: *mut *mut C_tensor,
13302 self_: *mut C_tensor,
13303 src_: *mut C_tensor,
13304 dim_: i64,
13305 index_: i64,
13306 );
13307 pub fn atg_select_scatter_out(
13308 out__: *mut *mut C_tensor,
13309 out_: *mut C_tensor,
13310 self_: *mut C_tensor,
13311 src_: *mut C_tensor,
13312 dim_: i64,
13313 index_: i64,
13314 );
13315 pub fn atg_selu(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13316 pub fn atg_selu_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13317 pub fn atg_set(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13318 pub fn atg_set_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13319 pub fn atg_set_data(self_: *mut C_tensor, new_data_: *mut C_tensor);
13320 pub fn atg_set_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13321 pub fn atg_set_requires_grad(out__: *mut *mut C_tensor, self_: *mut C_tensor, r_: c_int);
13322 pub fn atg_set_source_tensor(
13323 out__: *mut *mut C_tensor,
13324 self_: *mut C_tensor,
13325 source_: *mut C_tensor,
13326 );
13327 pub fn atg_set_source_tensor_(
13328 out__: *mut *mut C_tensor,
13329 self_: *mut C_tensor,
13330 source_: *mut C_tensor,
13331 );
13332 pub fn atg_set_source_tensor_out(
13333 out__: *mut *mut C_tensor,
13334 out_: *mut C_tensor,
13335 self_: *mut C_tensor,
13336 source_: *mut C_tensor,
13337 );
13338 pub fn atg_set_source_tensor_storage_offset_(
13339 out__: *mut *mut C_tensor,
13340 self_: *mut C_tensor,
13341 source_: *mut C_tensor,
13342 storage_offset_: i64,
13343 size_data: *const i64,
13344 size_len: c_int,
13345 stride_data: *const i64,
13346 stride_len: c_int,
13347 );
13348 pub fn atg_sgn(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13349 pub fn atg_sgn_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13350 pub fn atg_sgn_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13351 pub fn atg_sigmoid(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13352 pub fn atg_sigmoid_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13353 pub fn atg_sigmoid_backward(
13354 out__: *mut *mut C_tensor,
13355 grad_output_: *mut C_tensor,
13356 output_: *mut C_tensor,
13357 );
13358 pub fn atg_sigmoid_backward_grad_input(
13359 out__: *mut *mut C_tensor,
13360 grad_input_: *mut C_tensor,
13361 grad_output_: *mut C_tensor,
13362 output_: *mut C_tensor,
13363 );
13364 pub fn atg_sigmoid_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13365 pub fn atg_sign(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13366 pub fn atg_sign_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13367 pub fn atg_sign_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13368 pub fn atg_signbit(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13369 pub fn atg_signbit_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13370 pub fn atg_silu(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13371 pub fn atg_silu_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13372 pub fn atg_silu_backward(
13373 out__: *mut *mut C_tensor,
13374 grad_output_: *mut C_tensor,
13375 self_: *mut C_tensor,
13376 );
13377 pub fn atg_silu_backward_grad_input(
13378 out__: *mut *mut C_tensor,
13379 grad_input_: *mut C_tensor,
13380 grad_output_: *mut C_tensor,
13381 self_: *mut C_tensor,
13382 );
13383 pub fn atg_silu_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13384 pub fn atg_sin(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13385 pub fn atg_sin_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13386 pub fn atg_sin_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13387 pub fn atg_sinc(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13388 pub fn atg_sinc_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13389 pub fn atg_sinc_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13390 pub fn atg_sinh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13391 pub fn atg_sinh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13392 pub fn atg_sinh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13393 pub fn atg_slice(
13394 out__: *mut *mut C_tensor,
13395 self_: *mut C_tensor,
13396 dim_: i64,
13397 start_v: i64,
13398 start_null: i8,
13399 end_v: i64,
13400 end_null: i8,
13401 step_: i64,
13402 );
13403 pub fn atg_slice_backward(
13404 out__: *mut *mut C_tensor,
13405 grad_output_: *mut C_tensor,
13406 input_sizes_data: *const i64,
13407 input_sizes_len: c_int,
13408 dim_: i64,
13409 start_: i64,
13410 end_: i64,
13411 step_: i64,
13412 );
13413 pub fn atg_slice_backward_out(
13414 out__: *mut *mut C_tensor,
13415 out_: *mut C_tensor,
13416 grad_output_: *mut C_tensor,
13417 input_sizes_data: *const i64,
13418 input_sizes_len: c_int,
13419 dim_: i64,
13420 start_: i64,
13421 end_: i64,
13422 step_: i64,
13423 );
13424 pub fn atg_slice_copy(
13425 out__: *mut *mut C_tensor,
13426 self_: *mut C_tensor,
13427 dim_: i64,
13428 start_v: i64,
13429 start_null: i8,
13430 end_v: i64,
13431 end_null: i8,
13432 step_: i64,
13433 );
13434 pub fn atg_slice_copy_tensor_out(
13435 out__: *mut *mut C_tensor,
13436 out_: *mut C_tensor,
13437 self_: *mut C_tensor,
13438 dim_: i64,
13439 start_v: i64,
13440 start_null: i8,
13441 end_v: i64,
13442 end_null: i8,
13443 step_: i64,
13444 );
13445 pub fn atg_slice_inverse(
13446 out__: *mut *mut C_tensor,
13447 self_: *mut C_tensor,
13448 src_: *mut C_tensor,
13449 dim_: i64,
13450 start_v: i64,
13451 start_null: i8,
13452 end_v: i64,
13453 end_null: i8,
13454 step_: i64,
13455 );
13456 pub fn atg_slice_scatter(
13457 out__: *mut *mut C_tensor,
13458 self_: *mut C_tensor,
13459 src_: *mut C_tensor,
13460 dim_: i64,
13461 start_v: i64,
13462 start_null: i8,
13463 end_v: i64,
13464 end_null: i8,
13465 step_: i64,
13466 );
13467 pub fn atg_slice_scatter_out(
13468 out__: *mut *mut C_tensor,
13469 out_: *mut C_tensor,
13470 self_: *mut C_tensor,
13471 src_: *mut C_tensor,
13472 dim_: i64,
13473 start_v: i64,
13474 start_null: i8,
13475 end_v: i64,
13476 end_null: i8,
13477 step_: i64,
13478 );
13479 pub fn atg_slogdet(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13480 pub fn atg_slogdet_out(
13481 out__: *mut *mut C_tensor,
13482 sign_: *mut C_tensor,
13483 logabsdet_: *mut C_tensor,
13484 self_: *mut C_tensor,
13485 );
13486 pub fn atg_slow_conv3d(
13487 out__: *mut *mut C_tensor,
13488 self_: *mut C_tensor,
13489 weight_: *mut C_tensor,
13490 kernel_size_data: *const i64,
13491 kernel_size_len: c_int,
13492 bias_: *mut C_tensor,
13493 stride_data: *const i64,
13494 stride_len: c_int,
13495 padding_data: *const i64,
13496 padding_len: c_int,
13497 );
13498 pub fn atg_slow_conv3d_out(
13499 out__: *mut *mut C_tensor,
13500 out_: *mut C_tensor,
13501 self_: *mut C_tensor,
13502 weight_: *mut C_tensor,
13503 kernel_size_data: *const i64,
13504 kernel_size_len: c_int,
13505 bias_: *mut C_tensor,
13506 stride_data: *const i64,
13507 stride_len: c_int,
13508 padding_data: *const i64,
13509 padding_len: c_int,
13510 );
13511 pub fn atg_slow_conv_dilated2d(
13512 out__: *mut *mut C_tensor,
13513 self_: *mut C_tensor,
13514 weight_: *mut C_tensor,
13515 kernel_size_data: *const i64,
13516 kernel_size_len: c_int,
13517 bias_: *mut C_tensor,
13518 stride_data: *const i64,
13519 stride_len: c_int,
13520 padding_data: *const i64,
13521 padding_len: c_int,
13522 dilation_data: *const i64,
13523 dilation_len: c_int,
13524 );
13525 pub fn atg_slow_conv_dilated2d_out(
13526 out__: *mut *mut C_tensor,
13527 out_: *mut C_tensor,
13528 self_: *mut C_tensor,
13529 weight_: *mut C_tensor,
13530 kernel_size_data: *const i64,
13531 kernel_size_len: c_int,
13532 bias_: *mut C_tensor,
13533 stride_data: *const i64,
13534 stride_len: c_int,
13535 padding_data: *const i64,
13536 padding_len: c_int,
13537 dilation_data: *const i64,
13538 dilation_len: c_int,
13539 );
13540 pub fn atg_slow_conv_dilated3d(
13541 out__: *mut *mut C_tensor,
13542 self_: *mut C_tensor,
13543 weight_: *mut C_tensor,
13544 kernel_size_data: *const i64,
13545 kernel_size_len: c_int,
13546 bias_: *mut C_tensor,
13547 stride_data: *const i64,
13548 stride_len: c_int,
13549 padding_data: *const i64,
13550 padding_len: c_int,
13551 dilation_data: *const i64,
13552 dilation_len: c_int,
13553 );
13554 pub fn atg_slow_conv_dilated3d_out(
13555 out__: *mut *mut C_tensor,
13556 out_: *mut C_tensor,
13557 self_: *mut C_tensor,
13558 weight_: *mut C_tensor,
13559 kernel_size_data: *const i64,
13560 kernel_size_len: c_int,
13561 bias_: *mut C_tensor,
13562 stride_data: *const i64,
13563 stride_len: c_int,
13564 padding_data: *const i64,
13565 padding_len: c_int,
13566 dilation_data: *const i64,
13567 dilation_len: c_int,
13568 );
13569 pub fn atg_slow_conv_transpose2d(
13570 out__: *mut *mut C_tensor,
13571 self_: *mut C_tensor,
13572 weight_: *mut C_tensor,
13573 kernel_size_data: *const i64,
13574 kernel_size_len: c_int,
13575 bias_: *mut C_tensor,
13576 stride_data: *const i64,
13577 stride_len: c_int,
13578 padding_data: *const i64,
13579 padding_len: c_int,
13580 output_padding_data: *const i64,
13581 output_padding_len: c_int,
13582 dilation_data: *const i64,
13583 dilation_len: c_int,
13584 );
13585 pub fn atg_slow_conv_transpose2d_out(
13586 out__: *mut *mut C_tensor,
13587 out_: *mut C_tensor,
13588 self_: *mut C_tensor,
13589 weight_: *mut C_tensor,
13590 kernel_size_data: *const i64,
13591 kernel_size_len: c_int,
13592 bias_: *mut C_tensor,
13593 stride_data: *const i64,
13594 stride_len: c_int,
13595 padding_data: *const i64,
13596 padding_len: c_int,
13597 output_padding_data: *const i64,
13598 output_padding_len: c_int,
13599 dilation_data: *const i64,
13600 dilation_len: c_int,
13601 );
13602 pub fn atg_slow_conv_transpose3d(
13603 out__: *mut *mut C_tensor,
13604 self_: *mut C_tensor,
13605 weight_: *mut C_tensor,
13606 kernel_size_data: *const i64,
13607 kernel_size_len: c_int,
13608 bias_: *mut C_tensor,
13609 stride_data: *const i64,
13610 stride_len: c_int,
13611 padding_data: *const i64,
13612 padding_len: c_int,
13613 output_padding_data: *const i64,
13614 output_padding_len: c_int,
13615 dilation_data: *const i64,
13616 dilation_len: c_int,
13617 );
13618 pub fn atg_slow_conv_transpose3d_out(
13619 out__: *mut *mut C_tensor,
13620 out_: *mut C_tensor,
13621 self_: *mut C_tensor,
13622 weight_: *mut C_tensor,
13623 kernel_size_data: *const i64,
13624 kernel_size_len: c_int,
13625 bias_: *mut C_tensor,
13626 stride_data: *const i64,
13627 stride_len: c_int,
13628 padding_data: *const i64,
13629 padding_len: c_int,
13630 output_padding_data: *const i64,
13631 output_padding_len: c_int,
13632 dilation_data: *const i64,
13633 dilation_len: c_int,
13634 );
13635 pub fn atg_smm(out__: *mut *mut C_tensor, self_: *mut C_tensor, mat2_: *mut C_tensor);
13636 pub fn atg_smooth_l1_loss(
13637 out__: *mut *mut C_tensor,
13638 self_: *mut C_tensor,
13639 target_: *mut C_tensor,
13640 reduction_: i64,
13641 beta_: f64,
13642 );
13643 pub fn atg_smooth_l1_loss_backward(
13644 out__: *mut *mut C_tensor,
13645 grad_output_: *mut C_tensor,
13646 self_: *mut C_tensor,
13647 target_: *mut C_tensor,
13648 reduction_: i64,
13649 beta_: f64,
13650 );
13651 pub fn atg_smooth_l1_loss_backward_grad_input(
13652 out__: *mut *mut C_tensor,
13653 grad_input_: *mut C_tensor,
13654 grad_output_: *mut C_tensor,
13655 self_: *mut C_tensor,
13656 target_: *mut C_tensor,
13657 reduction_: i64,
13658 beta_: f64,
13659 );
13660 pub fn atg_smooth_l1_loss_out(
13661 out__: *mut *mut C_tensor,
13662 out_: *mut C_tensor,
13663 self_: *mut C_tensor,
13664 target_: *mut C_tensor,
13665 reduction_: i64,
13666 beta_: f64,
13667 );
13668 pub fn atg_soft_margin_loss(
13669 out__: *mut *mut C_tensor,
13670 self_: *mut C_tensor,
13671 target_: *mut C_tensor,
13672 reduction_: i64,
13673 );
13674 pub fn atg_soft_margin_loss_backward(
13675 out__: *mut *mut C_tensor,
13676 grad_output_: *mut C_tensor,
13677 self_: *mut C_tensor,
13678 target_: *mut C_tensor,
13679 reduction_: i64,
13680 );
13681 pub fn atg_soft_margin_loss_backward_grad_input(
13682 out__: *mut *mut C_tensor,
13683 grad_input_: *mut C_tensor,
13684 grad_output_: *mut C_tensor,
13685 self_: *mut C_tensor,
13686 target_: *mut C_tensor,
13687 reduction_: i64,
13688 );
13689 pub fn atg_soft_margin_loss_out(
13690 out__: *mut *mut C_tensor,
13691 out_: *mut C_tensor,
13692 self_: *mut C_tensor,
13693 target_: *mut C_tensor,
13694 reduction_: i64,
13695 );
13696 pub fn atg_softmax(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, dtype_: c_int);
13697 pub fn atg_softmax_int_out(
13698 out__: *mut *mut C_tensor,
13699 out_: *mut C_tensor,
13700 self_: *mut C_tensor,
13701 dim_: i64,
13702 dtype_: c_int,
13703 );
13704 pub fn atg_softplus(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13705 pub fn atg_softplus_backward(
13706 out__: *mut *mut C_tensor,
13707 grad_output_: *mut C_tensor,
13708 self_: *mut C_tensor,
13709 beta_: *mut C_scalar,
13710 threshold_: *mut C_scalar,
13711 );
13712 pub fn atg_softplus_backward_grad_input(
13713 out__: *mut *mut C_tensor,
13714 grad_input_: *mut C_tensor,
13715 grad_output_: *mut C_tensor,
13716 self_: *mut C_tensor,
13717 beta_: *mut C_scalar,
13718 threshold_: *mut C_scalar,
13719 );
13720 pub fn atg_softplus_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13721 pub fn atg_softshrink(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13722 pub fn atg_softshrink_backward(
13723 out__: *mut *mut C_tensor,
13724 grad_output_: *mut C_tensor,
13725 self_: *mut C_tensor,
13726 lambd_: *mut C_scalar,
13727 );
13728 pub fn atg_softshrink_backward_grad_input(
13729 out__: *mut *mut C_tensor,
13730 grad_input_: *mut C_tensor,
13731 grad_output_: *mut C_tensor,
13732 self_: *mut C_tensor,
13733 lambd_: *mut C_scalar,
13734 );
13735 pub fn atg_softshrink_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
13736 pub fn atg_sort(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64, descending_: c_int);
13737 pub fn atg_sort_stable(
13738 out__: *mut *mut C_tensor,
13739 self_: *mut C_tensor,
13740 stable_: c_int,
13741 dim_: i64,
13742 descending_: c_int,
13743 );
13744 pub fn atg_sort_values(
13745 out__: *mut *mut C_tensor,
13746 values_: *mut C_tensor,
13747 indices_: *mut C_tensor,
13748 self_: *mut C_tensor,
13749 dim_: i64,
13750 descending_: c_int,
13751 );
13752 pub fn atg_sort_values_stable(
13753 out__: *mut *mut C_tensor,
13754 values_: *mut C_tensor,
13755 indices_: *mut C_tensor,
13756 self_: *mut C_tensor,
13757 stable_: c_int,
13758 dim_: i64,
13759 descending_: c_int,
13760 );
13761 pub fn atg_sparse_bsc_tensor(
13762 out__: *mut *mut C_tensor,
13763 ccol_indices_: *mut C_tensor,
13764 row_indices_: *mut C_tensor,
13765 values_: *mut C_tensor,
13766 options_kind: c_int,
13767 options_device: c_int,
13768 );
13769 pub fn atg_sparse_bsc_tensor_ccol_row_value_size(
13770 out__: *mut *mut C_tensor,
13771 ccol_indices_: *mut C_tensor,
13772 row_indices_: *mut C_tensor,
13773 values_: *mut C_tensor,
13774 size_data: *const i64,
13775 size_len: c_int,
13776 options_kind: c_int,
13777 options_device: c_int,
13778 );
13779 pub fn atg_sparse_bsr_tensor(
13780 out__: *mut *mut C_tensor,
13781 crow_indices_: *mut C_tensor,
13782 col_indices_: *mut C_tensor,
13783 values_: *mut C_tensor,
13784 options_kind: c_int,
13785 options_device: c_int,
13786 );
13787 pub fn atg_sparse_bsr_tensor_crow_col_value_size(
13788 out__: *mut *mut C_tensor,
13789 crow_indices_: *mut C_tensor,
13790 col_indices_: *mut C_tensor,
13791 values_: *mut C_tensor,
13792 size_data: *const i64,
13793 size_len: c_int,
13794 options_kind: c_int,
13795 options_device: c_int,
13796 );
13797 pub fn atg_sparse_compressed_tensor(
13798 out__: *mut *mut C_tensor,
13799 compressed_indices_: *mut C_tensor,
13800 plain_indices_: *mut C_tensor,
13801 values_: *mut C_tensor,
13802 options_kind: c_int,
13803 options_device: c_int,
13804 );
13805 pub fn atg_sparse_compressed_tensor_comp_plain_value_size(
13806 out__: *mut *mut C_tensor,
13807 compressed_indices_: *mut C_tensor,
13808 plain_indices_: *mut C_tensor,
13809 values_: *mut C_tensor,
13810 size_data: *const i64,
13811 size_len: c_int,
13812 options_kind: c_int,
13813 options_device: c_int,
13814 );
13815 pub fn atg_sparse_coo_tensor(
13816 out__: *mut *mut C_tensor,
13817 size_data: *const i64,
13818 size_len: c_int,
13819 options_kind: c_int,
13820 options_device: c_int,
13821 );
13822 pub fn atg_sparse_coo_tensor_indices(
13823 out__: *mut *mut C_tensor,
13824 indices_: *mut C_tensor,
13825 values_: *mut C_tensor,
13826 options_kind: c_int,
13827 options_device: c_int,
13828 is_coalesced_: c_int,
13829 );
13830 pub fn atg_sparse_coo_tensor_indices_size(
13831 out__: *mut *mut C_tensor,
13832 indices_: *mut C_tensor,
13833 values_: *mut C_tensor,
13834 size_data: *const i64,
13835 size_len: c_int,
13836 options_kind: c_int,
13837 options_device: c_int,
13838 is_coalesced_: c_int,
13839 );
13840 pub fn atg_sparse_coo_tensor_size_out(
13841 out__: *mut *mut C_tensor,
13842 out_: *mut C_tensor,
13843 size_data: *const i64,
13844 size_len: c_int,
13845 );
13846 pub fn atg_sparse_csc_tensor(
13847 out__: *mut *mut C_tensor,
13848 ccol_indices_: *mut C_tensor,
13849 row_indices_: *mut C_tensor,
13850 values_: *mut C_tensor,
13851 options_kind: c_int,
13852 options_device: c_int,
13853 );
13854 pub fn atg_sparse_csc_tensor_ccol_row_value_size(
13855 out__: *mut *mut C_tensor,
13856 ccol_indices_: *mut C_tensor,
13857 row_indices_: *mut C_tensor,
13858 values_: *mut C_tensor,
13859 size_data: *const i64,
13860 size_len: c_int,
13861 options_kind: c_int,
13862 options_device: c_int,
13863 );
13864 pub fn atg_sparse_csr_tensor(
13865 out__: *mut *mut C_tensor,
13866 crow_indices_: *mut C_tensor,
13867 col_indices_: *mut C_tensor,
13868 values_: *mut C_tensor,
13869 options_kind: c_int,
13870 options_device: c_int,
13871 );
13872 pub fn atg_sparse_csr_tensor_crow_col_value_size(
13873 out__: *mut *mut C_tensor,
13874 crow_indices_: *mut C_tensor,
13875 col_indices_: *mut C_tensor,
13876 values_: *mut C_tensor,
13877 size_data: *const i64,
13878 size_len: c_int,
13879 options_kind: c_int,
13880 options_device: c_int,
13881 );
13882 pub fn atg_sparse_dim(self_: *mut C_tensor) -> i64;
13883 pub fn atg_sparse_mask(out__: *mut *mut C_tensor, self_: *mut C_tensor, mask_: *mut C_tensor);
13884 pub fn atg_sparse_mask_out(
13885 out__: *mut *mut C_tensor,
13886 out_: *mut C_tensor,
13887 self_: *mut C_tensor,
13888 mask_: *mut C_tensor,
13889 );
13890 pub fn atg_sparse_resize(
13891 out__: *mut *mut C_tensor,
13892 self_: *mut C_tensor,
13893 size_data: *const i64,
13894 size_len: c_int,
13895 sparse_dim_: i64,
13896 dense_dim_: i64,
13897 );
13898 pub fn atg_sparse_resize_(
13899 out__: *mut *mut C_tensor,
13900 self_: *mut C_tensor,
13901 size_data: *const i64,
13902 size_len: c_int,
13903 sparse_dim_: i64,
13904 dense_dim_: i64,
13905 );
13906 pub fn atg_sparse_resize_and_clear(
13907 out__: *mut *mut C_tensor,
13908 self_: *mut C_tensor,
13909 size_data: *const i64,
13910 size_len: c_int,
13911 sparse_dim_: i64,
13912 dense_dim_: i64,
13913 );
13914 pub fn atg_sparse_resize_and_clear_(
13915 out__: *mut *mut C_tensor,
13916 self_: *mut C_tensor,
13917 size_data: *const i64,
13918 size_len: c_int,
13919 sparse_dim_: i64,
13920 dense_dim_: i64,
13921 );
13922 pub fn atg_sparse_resize_and_clear_out(
13923 out__: *mut *mut C_tensor,
13924 out_: *mut C_tensor,
13925 self_: *mut C_tensor,
13926 size_data: *const i64,
13927 size_len: c_int,
13928 sparse_dim_: i64,
13929 dense_dim_: i64,
13930 );
13931 pub fn atg_sparse_resize_out(
13932 out__: *mut *mut C_tensor,
13933 out_: *mut C_tensor,
13934 self_: *mut C_tensor,
13935 size_data: *const i64,
13936 size_len: c_int,
13937 sparse_dim_: i64,
13938 dense_dim_: i64,
13939 );
13940 pub fn atg_sparse_sampled_addmm(
13941 out__: *mut *mut C_tensor,
13942 self_: *mut C_tensor,
13943 mat1_: *mut C_tensor,
13944 mat2_: *mut C_tensor,
13945 );
13946 pub fn atg_sparse_sampled_addmm_out(
13947 out__: *mut *mut C_tensor,
13948 out_: *mut C_tensor,
13949 self_: *mut C_tensor,
13950 mat1_: *mut C_tensor,
13951 mat2_: *mut C_tensor,
13952 );
13953 pub fn atg_special_airy_ai(out__: *mut *mut C_tensor, x_: *mut C_tensor);
13954 pub fn atg_special_airy_ai_out(
13955 out__: *mut *mut C_tensor,
13956 out_: *mut C_tensor,
13957 x_: *mut C_tensor,
13958 );
13959 pub fn atg_special_bessel_j0(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13960 pub fn atg_special_bessel_j0_out(
13961 out__: *mut *mut C_tensor,
13962 out_: *mut C_tensor,
13963 self_: *mut C_tensor,
13964 );
13965 pub fn atg_special_bessel_j1(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13966 pub fn atg_special_bessel_j1_out(
13967 out__: *mut *mut C_tensor,
13968 out_: *mut C_tensor,
13969 self_: *mut C_tensor,
13970 );
13971 pub fn atg_special_bessel_y0(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13972 pub fn atg_special_bessel_y0_out(
13973 out__: *mut *mut C_tensor,
13974 out_: *mut C_tensor,
13975 self_: *mut C_tensor,
13976 );
13977 pub fn atg_special_bessel_y1(out__: *mut *mut C_tensor, self_: *mut C_tensor);
13978 pub fn atg_special_bessel_y1_out(
13979 out__: *mut *mut C_tensor,
13980 out_: *mut C_tensor,
13981 self_: *mut C_tensor,
13982 );
13983 pub fn atg_special_chebyshev_polynomial_t(
13984 out__: *mut *mut C_tensor,
13985 x_: *mut C_tensor,
13986 n_: *mut C_tensor,
13987 );
13988 pub fn atg_special_chebyshev_polynomial_t_n_scalar(
13989 out__: *mut *mut C_tensor,
13990 x_: *mut C_tensor,
13991 n_: *mut C_scalar,
13992 );
13993 pub fn atg_special_chebyshev_polynomial_t_n_scalar_out(
13994 out__: *mut *mut C_tensor,
13995 out_: *mut C_tensor,
13996 x_: *mut C_tensor,
13997 n_: *mut C_scalar,
13998 );
13999 pub fn atg_special_chebyshev_polynomial_t_out(
14000 out__: *mut *mut C_tensor,
14001 out_: *mut C_tensor,
14002 x_: *mut C_tensor,
14003 n_: *mut C_tensor,
14004 );
14005 pub fn atg_special_chebyshev_polynomial_t_x_scalar(
14006 out__: *mut *mut C_tensor,
14007 x_: *mut C_scalar,
14008 n_: *mut C_tensor,
14009 );
14010 pub fn atg_special_chebyshev_polynomial_t_x_scalar_out(
14011 out__: *mut *mut C_tensor,
14012 out_: *mut C_tensor,
14013 x_: *mut C_scalar,
14014 n_: *mut C_tensor,
14015 );
14016 pub fn atg_special_chebyshev_polynomial_u(
14017 out__: *mut *mut C_tensor,
14018 x_: *mut C_tensor,
14019 n_: *mut C_tensor,
14020 );
14021 pub fn atg_special_chebyshev_polynomial_u_n_scalar(
14022 out__: *mut *mut C_tensor,
14023 x_: *mut C_tensor,
14024 n_: *mut C_scalar,
14025 );
14026 pub fn atg_special_chebyshev_polynomial_u_n_scalar_out(
14027 out__: *mut *mut C_tensor,
14028 out_: *mut C_tensor,
14029 x_: *mut C_tensor,
14030 n_: *mut C_scalar,
14031 );
14032 pub fn atg_special_chebyshev_polynomial_u_out(
14033 out__: *mut *mut C_tensor,
14034 out_: *mut C_tensor,
14035 x_: *mut C_tensor,
14036 n_: *mut C_tensor,
14037 );
14038 pub fn atg_special_chebyshev_polynomial_u_x_scalar(
14039 out__: *mut *mut C_tensor,
14040 x_: *mut C_scalar,
14041 n_: *mut C_tensor,
14042 );
14043 pub fn atg_special_chebyshev_polynomial_u_x_scalar_out(
14044 out__: *mut *mut C_tensor,
14045 out_: *mut C_tensor,
14046 x_: *mut C_scalar,
14047 n_: *mut C_tensor,
14048 );
14049 pub fn atg_special_chebyshev_polynomial_v(
14050 out__: *mut *mut C_tensor,
14051 x_: *mut C_tensor,
14052 n_: *mut C_tensor,
14053 );
14054 pub fn atg_special_chebyshev_polynomial_v_n_scalar(
14055 out__: *mut *mut C_tensor,
14056 x_: *mut C_tensor,
14057 n_: *mut C_scalar,
14058 );
14059 pub fn atg_special_chebyshev_polynomial_v_n_scalar_out(
14060 out__: *mut *mut C_tensor,
14061 out_: *mut C_tensor,
14062 x_: *mut C_tensor,
14063 n_: *mut C_scalar,
14064 );
14065 pub fn atg_special_chebyshev_polynomial_v_out(
14066 out__: *mut *mut C_tensor,
14067 out_: *mut C_tensor,
14068 x_: *mut C_tensor,
14069 n_: *mut C_tensor,
14070 );
14071 pub fn atg_special_chebyshev_polynomial_v_x_scalar(
14072 out__: *mut *mut C_tensor,
14073 x_: *mut C_scalar,
14074 n_: *mut C_tensor,
14075 );
14076 pub fn atg_special_chebyshev_polynomial_v_x_scalar_out(
14077 out__: *mut *mut C_tensor,
14078 out_: *mut C_tensor,
14079 x_: *mut C_scalar,
14080 n_: *mut C_tensor,
14081 );
14082 pub fn atg_special_chebyshev_polynomial_w(
14083 out__: *mut *mut C_tensor,
14084 x_: *mut C_tensor,
14085 n_: *mut C_tensor,
14086 );
14087 pub fn atg_special_chebyshev_polynomial_w_n_scalar(
14088 out__: *mut *mut C_tensor,
14089 x_: *mut C_tensor,
14090 n_: *mut C_scalar,
14091 );
14092 pub fn atg_special_chebyshev_polynomial_w_n_scalar_out(
14093 out__: *mut *mut C_tensor,
14094 out_: *mut C_tensor,
14095 x_: *mut C_tensor,
14096 n_: *mut C_scalar,
14097 );
14098 pub fn atg_special_chebyshev_polynomial_w_out(
14099 out__: *mut *mut C_tensor,
14100 out_: *mut C_tensor,
14101 x_: *mut C_tensor,
14102 n_: *mut C_tensor,
14103 );
14104 pub fn atg_special_chebyshev_polynomial_w_x_scalar(
14105 out__: *mut *mut C_tensor,
14106 x_: *mut C_scalar,
14107 n_: *mut C_tensor,
14108 );
14109 pub fn atg_special_chebyshev_polynomial_w_x_scalar_out(
14110 out__: *mut *mut C_tensor,
14111 out_: *mut C_tensor,
14112 x_: *mut C_scalar,
14113 n_: *mut C_tensor,
14114 );
14115 pub fn atg_special_digamma(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14116 pub fn atg_special_digamma_out(
14117 out__: *mut *mut C_tensor,
14118 out_: *mut C_tensor,
14119 self_: *mut C_tensor,
14120 );
14121 pub fn atg_special_entr(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14122 pub fn atg_special_entr_out(
14123 out__: *mut *mut C_tensor,
14124 out_: *mut C_tensor,
14125 self_: *mut C_tensor,
14126 );
14127 pub fn atg_special_erf(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14128 pub fn atg_special_erf_out(
14129 out__: *mut *mut C_tensor,
14130 out_: *mut C_tensor,
14131 self_: *mut C_tensor,
14132 );
14133 pub fn atg_special_erfc(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14134 pub fn atg_special_erfc_out(
14135 out__: *mut *mut C_tensor,
14136 out_: *mut C_tensor,
14137 self_: *mut C_tensor,
14138 );
14139 pub fn atg_special_erfcx(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14140 pub fn atg_special_erfcx_out(
14141 out__: *mut *mut C_tensor,
14142 out_: *mut C_tensor,
14143 self_: *mut C_tensor,
14144 );
14145 pub fn atg_special_erfinv(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14146 pub fn atg_special_erfinv_out(
14147 out__: *mut *mut C_tensor,
14148 out_: *mut C_tensor,
14149 self_: *mut C_tensor,
14150 );
14151 pub fn atg_special_exp2(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14152 pub fn atg_special_exp2_out(
14153 out__: *mut *mut C_tensor,
14154 out_: *mut C_tensor,
14155 self_: *mut C_tensor,
14156 );
14157 pub fn atg_special_expit(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14158 pub fn atg_special_expit_out(
14159 out__: *mut *mut C_tensor,
14160 out_: *mut C_tensor,
14161 self_: *mut C_tensor,
14162 );
14163 pub fn atg_special_expm1(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14164 pub fn atg_special_expm1_out(
14165 out__: *mut *mut C_tensor,
14166 out_: *mut C_tensor,
14167 self_: *mut C_tensor,
14168 );
14169 pub fn atg_special_gammainc(
14170 out__: *mut *mut C_tensor,
14171 self_: *mut C_tensor,
14172 other_: *mut C_tensor,
14173 );
14174 pub fn atg_special_gammainc_out(
14175 out__: *mut *mut C_tensor,
14176 out_: *mut C_tensor,
14177 self_: *mut C_tensor,
14178 other_: *mut C_tensor,
14179 );
14180 pub fn atg_special_gammaincc(
14181 out__: *mut *mut C_tensor,
14182 self_: *mut C_tensor,
14183 other_: *mut C_tensor,
14184 );
14185 pub fn atg_special_gammaincc_out(
14186 out__: *mut *mut C_tensor,
14187 out_: *mut C_tensor,
14188 self_: *mut C_tensor,
14189 other_: *mut C_tensor,
14190 );
14191 pub fn atg_special_gammaln(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14192 pub fn atg_special_gammaln_out(
14193 out__: *mut *mut C_tensor,
14194 out_: *mut C_tensor,
14195 self_: *mut C_tensor,
14196 );
14197 pub fn atg_special_hermite_polynomial_h(
14198 out__: *mut *mut C_tensor,
14199 x_: *mut C_tensor,
14200 n_: *mut C_tensor,
14201 );
14202 pub fn atg_special_hermite_polynomial_h_n_scalar(
14203 out__: *mut *mut C_tensor,
14204 x_: *mut C_tensor,
14205 n_: *mut C_scalar,
14206 );
14207 pub fn atg_special_hermite_polynomial_h_n_scalar_out(
14208 out__: *mut *mut C_tensor,
14209 out_: *mut C_tensor,
14210 x_: *mut C_tensor,
14211 n_: *mut C_scalar,
14212 );
14213 pub fn atg_special_hermite_polynomial_h_out(
14214 out__: *mut *mut C_tensor,
14215 out_: *mut C_tensor,
14216 x_: *mut C_tensor,
14217 n_: *mut C_tensor,
14218 );
14219 pub fn atg_special_hermite_polynomial_h_x_scalar(
14220 out__: *mut *mut C_tensor,
14221 x_: *mut C_scalar,
14222 n_: *mut C_tensor,
14223 );
14224 pub fn atg_special_hermite_polynomial_h_x_scalar_out(
14225 out__: *mut *mut C_tensor,
14226 out_: *mut C_tensor,
14227 x_: *mut C_scalar,
14228 n_: *mut C_tensor,
14229 );
14230 pub fn atg_special_hermite_polynomial_he(
14231 out__: *mut *mut C_tensor,
14232 x_: *mut C_tensor,
14233 n_: *mut C_tensor,
14234 );
14235 pub fn atg_special_hermite_polynomial_he_n_scalar(
14236 out__: *mut *mut C_tensor,
14237 x_: *mut C_tensor,
14238 n_: *mut C_scalar,
14239 );
14240 pub fn atg_special_hermite_polynomial_he_n_scalar_out(
14241 out__: *mut *mut C_tensor,
14242 out_: *mut C_tensor,
14243 x_: *mut C_tensor,
14244 n_: *mut C_scalar,
14245 );
14246 pub fn atg_special_hermite_polynomial_he_out(
14247 out__: *mut *mut C_tensor,
14248 out_: *mut C_tensor,
14249 x_: *mut C_tensor,
14250 n_: *mut C_tensor,
14251 );
14252 pub fn atg_special_hermite_polynomial_he_x_scalar(
14253 out__: *mut *mut C_tensor,
14254 x_: *mut C_scalar,
14255 n_: *mut C_tensor,
14256 );
14257 pub fn atg_special_hermite_polynomial_he_x_scalar_out(
14258 out__: *mut *mut C_tensor,
14259 out_: *mut C_tensor,
14260 x_: *mut C_scalar,
14261 n_: *mut C_tensor,
14262 );
14263 pub fn atg_special_i0(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14264 pub fn atg_special_i0_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
14265 pub fn atg_special_i0e(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14266 pub fn atg_special_i0e_out(
14267 out__: *mut *mut C_tensor,
14268 out_: *mut C_tensor,
14269 self_: *mut C_tensor,
14270 );
14271 pub fn atg_special_i1(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14272 pub fn atg_special_i1_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
14273 pub fn atg_special_i1e(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14274 pub fn atg_special_i1e_out(
14275 out__: *mut *mut C_tensor,
14276 out_: *mut C_tensor,
14277 self_: *mut C_tensor,
14278 );
14279 pub fn atg_special_laguerre_polynomial_l(
14280 out__: *mut *mut C_tensor,
14281 x_: *mut C_tensor,
14282 n_: *mut C_tensor,
14283 );
14284 pub fn atg_special_laguerre_polynomial_l_n_scalar(
14285 out__: *mut *mut C_tensor,
14286 x_: *mut C_tensor,
14287 n_: *mut C_scalar,
14288 );
14289 pub fn atg_special_laguerre_polynomial_l_n_scalar_out(
14290 out__: *mut *mut C_tensor,
14291 out_: *mut C_tensor,
14292 x_: *mut C_tensor,
14293 n_: *mut C_scalar,
14294 );
14295 pub fn atg_special_laguerre_polynomial_l_out(
14296 out__: *mut *mut C_tensor,
14297 out_: *mut C_tensor,
14298 x_: *mut C_tensor,
14299 n_: *mut C_tensor,
14300 );
14301 pub fn atg_special_laguerre_polynomial_l_x_scalar(
14302 out__: *mut *mut C_tensor,
14303 x_: *mut C_scalar,
14304 n_: *mut C_tensor,
14305 );
14306 pub fn atg_special_laguerre_polynomial_l_x_scalar_out(
14307 out__: *mut *mut C_tensor,
14308 out_: *mut C_tensor,
14309 x_: *mut C_scalar,
14310 n_: *mut C_tensor,
14311 );
14312 pub fn atg_special_legendre_polynomial_p(
14313 out__: *mut *mut C_tensor,
14314 x_: *mut C_tensor,
14315 n_: *mut C_tensor,
14316 );
14317 pub fn atg_special_legendre_polynomial_p_n_scalar(
14318 out__: *mut *mut C_tensor,
14319 x_: *mut C_tensor,
14320 n_: *mut C_scalar,
14321 );
14322 pub fn atg_special_legendre_polynomial_p_n_scalar_out(
14323 out__: *mut *mut C_tensor,
14324 out_: *mut C_tensor,
14325 x_: *mut C_tensor,
14326 n_: *mut C_scalar,
14327 );
14328 pub fn atg_special_legendre_polynomial_p_out(
14329 out__: *mut *mut C_tensor,
14330 out_: *mut C_tensor,
14331 x_: *mut C_tensor,
14332 n_: *mut C_tensor,
14333 );
14334 pub fn atg_special_legendre_polynomial_p_x_scalar(
14335 out__: *mut *mut C_tensor,
14336 x_: *mut C_scalar,
14337 n_: *mut C_tensor,
14338 );
14339 pub fn atg_special_legendre_polynomial_p_x_scalar_out(
14340 out__: *mut *mut C_tensor,
14341 out_: *mut C_tensor,
14342 x_: *mut C_scalar,
14343 n_: *mut C_tensor,
14344 );
14345 pub fn atg_special_log1p(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14346 pub fn atg_special_log1p_out(
14347 out__: *mut *mut C_tensor,
14348 out_: *mut C_tensor,
14349 self_: *mut C_tensor,
14350 );
14351 pub fn atg_special_log_ndtr(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14352 pub fn atg_special_log_ndtr_out(
14353 out__: *mut *mut C_tensor,
14354 out_: *mut C_tensor,
14355 self_: *mut C_tensor,
14356 );
14357 pub fn atg_special_log_softmax(
14358 out__: *mut *mut C_tensor,
14359 self_: *mut C_tensor,
14360 dim_: i64,
14361 dtype_: c_int,
14362 );
14363 pub fn atg_special_logit(
14364 out__: *mut *mut C_tensor,
14365 self_: *mut C_tensor,
14366 eps_v: f64,
14367 eps_null: i8,
14368 );
14369 pub fn atg_special_logit_out(
14370 out__: *mut *mut C_tensor,
14371 out_: *mut C_tensor,
14372 self_: *mut C_tensor,
14373 eps_v: f64,
14374 eps_null: i8,
14375 );
14376 pub fn atg_special_logsumexp(
14377 out__: *mut *mut C_tensor,
14378 self_: *mut C_tensor,
14379 dim_data: *const i64,
14380 dim_len: c_int,
14381 keepdim_: c_int,
14382 );
14383 pub fn atg_special_logsumexp_out(
14384 out__: *mut *mut C_tensor,
14385 out_: *mut C_tensor,
14386 self_: *mut C_tensor,
14387 dim_data: *const i64,
14388 dim_len: c_int,
14389 keepdim_: c_int,
14390 );
14391 pub fn atg_special_modified_bessel_i0(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14392 pub fn atg_special_modified_bessel_i0_out(
14393 out__: *mut *mut C_tensor,
14394 out_: *mut C_tensor,
14395 self_: *mut C_tensor,
14396 );
14397 pub fn atg_special_modified_bessel_i1(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14398 pub fn atg_special_modified_bessel_i1_out(
14399 out__: *mut *mut C_tensor,
14400 out_: *mut C_tensor,
14401 self_: *mut C_tensor,
14402 );
14403 pub fn atg_special_modified_bessel_k0(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14404 pub fn atg_special_modified_bessel_k0_out(
14405 out__: *mut *mut C_tensor,
14406 out_: *mut C_tensor,
14407 self_: *mut C_tensor,
14408 );
14409 pub fn atg_special_modified_bessel_k1(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14410 pub fn atg_special_modified_bessel_k1_out(
14411 out__: *mut *mut C_tensor,
14412 out_: *mut C_tensor,
14413 self_: *mut C_tensor,
14414 );
14415 pub fn atg_special_multigammaln(out__: *mut *mut C_tensor, self_: *mut C_tensor, p_: i64);
14416 pub fn atg_special_multigammaln_out(
14417 out__: *mut *mut C_tensor,
14418 out_: *mut C_tensor,
14419 self_: *mut C_tensor,
14420 p_: i64,
14421 );
14422 pub fn atg_special_ndtr(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14423 pub fn atg_special_ndtr_out(
14424 out__: *mut *mut C_tensor,
14425 out_: *mut C_tensor,
14426 self_: *mut C_tensor,
14427 );
14428 pub fn atg_special_ndtri(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14429 pub fn atg_special_ndtri_out(
14430 out__: *mut *mut C_tensor,
14431 out_: *mut C_tensor,
14432 self_: *mut C_tensor,
14433 );
14434 pub fn atg_special_polygamma(out__: *mut *mut C_tensor, n_: i64, self_: *mut C_tensor);
14435 pub fn atg_special_polygamma_out(
14436 out__: *mut *mut C_tensor,
14437 out_: *mut C_tensor,
14438 n_: i64,
14439 self_: *mut C_tensor,
14440 );
14441 pub fn atg_special_psi(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14442 pub fn atg_special_psi_out(
14443 out__: *mut *mut C_tensor,
14444 out_: *mut C_tensor,
14445 self_: *mut C_tensor,
14446 );
14447 pub fn atg_special_round(out__: *mut *mut C_tensor, self_: *mut C_tensor, decimals_: i64);
14448 pub fn atg_special_round_out(
14449 out__: *mut *mut C_tensor,
14450 out_: *mut C_tensor,
14451 self_: *mut C_tensor,
14452 decimals_: i64,
14453 );
14454 pub fn atg_special_scaled_modified_bessel_k0(out__: *mut *mut C_tensor, x_: *mut C_tensor);
14455 pub fn atg_special_scaled_modified_bessel_k0_out(
14456 out__: *mut *mut C_tensor,
14457 out_: *mut C_tensor,
14458 x_: *mut C_tensor,
14459 );
14460 pub fn atg_special_scaled_modified_bessel_k1(out__: *mut *mut C_tensor, x_: *mut C_tensor);
14461 pub fn atg_special_scaled_modified_bessel_k1_out(
14462 out__: *mut *mut C_tensor,
14463 out_: *mut C_tensor,
14464 x_: *mut C_tensor,
14465 );
14466 pub fn atg_special_shifted_chebyshev_polynomial_t(
14467 out__: *mut *mut C_tensor,
14468 x_: *mut C_tensor,
14469 n_: *mut C_tensor,
14470 );
14471 pub fn atg_special_shifted_chebyshev_polynomial_t_n_scalar(
14472 out__: *mut *mut C_tensor,
14473 x_: *mut C_tensor,
14474 n_: *mut C_scalar,
14475 );
14476 pub fn atg_special_shifted_chebyshev_polynomial_t_n_scalar_out(
14477 out__: *mut *mut C_tensor,
14478 out_: *mut C_tensor,
14479 x_: *mut C_tensor,
14480 n_: *mut C_scalar,
14481 );
14482 pub fn atg_special_shifted_chebyshev_polynomial_t_out(
14483 out__: *mut *mut C_tensor,
14484 out_: *mut C_tensor,
14485 x_: *mut C_tensor,
14486 n_: *mut C_tensor,
14487 );
14488 pub fn atg_special_shifted_chebyshev_polynomial_t_x_scalar(
14489 out__: *mut *mut C_tensor,
14490 x_: *mut C_scalar,
14491 n_: *mut C_tensor,
14492 );
14493 pub fn atg_special_shifted_chebyshev_polynomial_t_x_scalar_out(
14494 out__: *mut *mut C_tensor,
14495 out_: *mut C_tensor,
14496 x_: *mut C_scalar,
14497 n_: *mut C_tensor,
14498 );
14499 pub fn atg_special_shifted_chebyshev_polynomial_u(
14500 out__: *mut *mut C_tensor,
14501 x_: *mut C_tensor,
14502 n_: *mut C_tensor,
14503 );
14504 pub fn atg_special_shifted_chebyshev_polynomial_u_n_scalar(
14505 out__: *mut *mut C_tensor,
14506 x_: *mut C_tensor,
14507 n_: *mut C_scalar,
14508 );
14509 pub fn atg_special_shifted_chebyshev_polynomial_u_n_scalar_out(
14510 out__: *mut *mut C_tensor,
14511 out_: *mut C_tensor,
14512 x_: *mut C_tensor,
14513 n_: *mut C_scalar,
14514 );
14515 pub fn atg_special_shifted_chebyshev_polynomial_u_out(
14516 out__: *mut *mut C_tensor,
14517 out_: *mut C_tensor,
14518 x_: *mut C_tensor,
14519 n_: *mut C_tensor,
14520 );
14521 pub fn atg_special_shifted_chebyshev_polynomial_u_x_scalar(
14522 out__: *mut *mut C_tensor,
14523 x_: *mut C_scalar,
14524 n_: *mut C_tensor,
14525 );
14526 pub fn atg_special_shifted_chebyshev_polynomial_u_x_scalar_out(
14527 out__: *mut *mut C_tensor,
14528 out_: *mut C_tensor,
14529 x_: *mut C_scalar,
14530 n_: *mut C_tensor,
14531 );
14532 pub fn atg_special_shifted_chebyshev_polynomial_v(
14533 out__: *mut *mut C_tensor,
14534 x_: *mut C_tensor,
14535 n_: *mut C_tensor,
14536 );
14537 pub fn atg_special_shifted_chebyshev_polynomial_v_n_scalar(
14538 out__: *mut *mut C_tensor,
14539 x_: *mut C_tensor,
14540 n_: *mut C_scalar,
14541 );
14542 pub fn atg_special_shifted_chebyshev_polynomial_v_n_scalar_out(
14543 out__: *mut *mut C_tensor,
14544 out_: *mut C_tensor,
14545 x_: *mut C_tensor,
14546 n_: *mut C_scalar,
14547 );
14548 pub fn atg_special_shifted_chebyshev_polynomial_v_out(
14549 out__: *mut *mut C_tensor,
14550 out_: *mut C_tensor,
14551 x_: *mut C_tensor,
14552 n_: *mut C_tensor,
14553 );
14554 pub fn atg_special_shifted_chebyshev_polynomial_v_x_scalar(
14555 out__: *mut *mut C_tensor,
14556 x_: *mut C_scalar,
14557 n_: *mut C_tensor,
14558 );
14559 pub fn atg_special_shifted_chebyshev_polynomial_v_x_scalar_out(
14560 out__: *mut *mut C_tensor,
14561 out_: *mut C_tensor,
14562 x_: *mut C_scalar,
14563 n_: *mut C_tensor,
14564 );
14565 pub fn atg_special_shifted_chebyshev_polynomial_w(
14566 out__: *mut *mut C_tensor,
14567 x_: *mut C_tensor,
14568 n_: *mut C_tensor,
14569 );
14570 pub fn atg_special_shifted_chebyshev_polynomial_w_n_scalar(
14571 out__: *mut *mut C_tensor,
14572 x_: *mut C_tensor,
14573 n_: *mut C_scalar,
14574 );
14575 pub fn atg_special_shifted_chebyshev_polynomial_w_n_scalar_out(
14576 out__: *mut *mut C_tensor,
14577 out_: *mut C_tensor,
14578 x_: *mut C_tensor,
14579 n_: *mut C_scalar,
14580 );
14581 pub fn atg_special_shifted_chebyshev_polynomial_w_out(
14582 out__: *mut *mut C_tensor,
14583 out_: *mut C_tensor,
14584 x_: *mut C_tensor,
14585 n_: *mut C_tensor,
14586 );
14587 pub fn atg_special_shifted_chebyshev_polynomial_w_x_scalar(
14588 out__: *mut *mut C_tensor,
14589 x_: *mut C_scalar,
14590 n_: *mut C_tensor,
14591 );
14592 pub fn atg_special_shifted_chebyshev_polynomial_w_x_scalar_out(
14593 out__: *mut *mut C_tensor,
14594 out_: *mut C_tensor,
14595 x_: *mut C_scalar,
14596 n_: *mut C_tensor,
14597 );
14598 pub fn atg_special_sinc(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14599 pub fn atg_special_sinc_out(
14600 out__: *mut *mut C_tensor,
14601 out_: *mut C_tensor,
14602 self_: *mut C_tensor,
14603 );
14604 pub fn atg_special_softmax(
14605 out__: *mut *mut C_tensor,
14606 self_: *mut C_tensor,
14607 dim_: i64,
14608 dtype_: c_int,
14609 );
14610 pub fn atg_special_spherical_bessel_j0(out__: *mut *mut C_tensor, x_: *mut C_tensor);
14611 pub fn atg_special_spherical_bessel_j0_out(
14612 out__: *mut *mut C_tensor,
14613 out_: *mut C_tensor,
14614 x_: *mut C_tensor,
14615 );
14616 pub fn atg_special_xlog1py(
14617 out__: *mut *mut C_tensor,
14618 self_: *mut C_tensor,
14619 other_: *mut C_tensor,
14620 );
14621 pub fn atg_special_xlog1py_other_scalar(
14622 out__: *mut *mut C_tensor,
14623 self_: *mut C_tensor,
14624 other_: *mut C_scalar,
14625 );
14626 pub fn atg_special_xlog1py_other_scalar_out(
14627 out__: *mut *mut C_tensor,
14628 out_: *mut C_tensor,
14629 self_: *mut C_tensor,
14630 other_: *mut C_scalar,
14631 );
14632 pub fn atg_special_xlog1py_out(
14633 out__: *mut *mut C_tensor,
14634 out_: *mut C_tensor,
14635 self_: *mut C_tensor,
14636 other_: *mut C_tensor,
14637 );
14638 pub fn atg_special_xlog1py_self_scalar(
14639 out__: *mut *mut C_tensor,
14640 self_scalar_: *mut C_scalar,
14641 other_: *mut C_tensor,
14642 );
14643 pub fn atg_special_xlog1py_self_scalar_out(
14644 out__: *mut *mut C_tensor,
14645 out_: *mut C_tensor,
14646 self_scalar_: *mut C_scalar,
14647 other_: *mut C_tensor,
14648 );
14649 pub fn atg_special_xlogy(
14650 out__: *mut *mut C_tensor,
14651 self_: *mut C_tensor,
14652 other_: *mut C_tensor,
14653 );
14654 pub fn atg_special_xlogy_other_scalar(
14655 out__: *mut *mut C_tensor,
14656 self_: *mut C_tensor,
14657 other_: *mut C_scalar,
14658 );
14659 pub fn atg_special_xlogy_other_scalar_out(
14660 out__: *mut *mut C_tensor,
14661 out_: *mut C_tensor,
14662 self_: *mut C_tensor,
14663 other_: *mut C_scalar,
14664 );
14665 pub fn atg_special_xlogy_out(
14666 out__: *mut *mut C_tensor,
14667 out_: *mut C_tensor,
14668 self_: *mut C_tensor,
14669 other_: *mut C_tensor,
14670 );
14671 pub fn atg_special_xlogy_self_scalar(
14672 out__: *mut *mut C_tensor,
14673 self_scalar_: *mut C_scalar,
14674 other_: *mut C_tensor,
14675 );
14676 pub fn atg_special_xlogy_self_scalar_out(
14677 out__: *mut *mut C_tensor,
14678 out_: *mut C_tensor,
14679 self_scalar_: *mut C_scalar,
14680 other_: *mut C_tensor,
14681 );
14682 pub fn atg_special_zeta(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
14683 pub fn atg_special_zeta_other_scalar(
14684 out__: *mut *mut C_tensor,
14685 self_: *mut C_tensor,
14686 other_: *mut C_scalar,
14687 );
14688 pub fn atg_special_zeta_other_scalar_out(
14689 out__: *mut *mut C_tensor,
14690 out_: *mut C_tensor,
14691 self_: *mut C_tensor,
14692 other_: *mut C_scalar,
14693 );
14694 pub fn atg_special_zeta_out(
14695 out__: *mut *mut C_tensor,
14696 out_: *mut C_tensor,
14697 self_: *mut C_tensor,
14698 other_: *mut C_tensor,
14699 );
14700 pub fn atg_special_zeta_self_scalar(
14701 out__: *mut *mut C_tensor,
14702 self_scalar_: *mut C_scalar,
14703 other_: *mut C_tensor,
14704 );
14705 pub fn atg_special_zeta_self_scalar_out(
14706 out__: *mut *mut C_tensor,
14707 out_: *mut C_tensor,
14708 self_scalar_: *mut C_scalar,
14709 other_: *mut C_tensor,
14710 );
14711 pub fn atg_split(self_: *mut C_tensor, split_size_: i64, dim_: i64) -> *mut *mut C_tensor;
14712 pub fn atg_split_copy(self_: *mut C_tensor, split_size_: i64, dim_: i64) -> *mut *mut C_tensor;
14713 pub fn atg_split_copy_tensor_out(
14714 out_data: *const *mut C_tensor,
14715 out_len: c_int,
14716 self_: *mut C_tensor,
14717 split_size_: i64,
14718 dim_: i64,
14719 );
14720 pub fn atg_split_sizes(
14721 self_: *mut C_tensor,
14722 split_size_data: *const i64,
14723 split_size_len: c_int,
14724 dim_: i64,
14725 ) -> *mut *mut C_tensor;
14726 pub fn atg_split_with_sizes(
14727 self_: *mut C_tensor,
14728 split_sizes_data: *const i64,
14729 split_sizes_len: c_int,
14730 dim_: i64,
14731 ) -> *mut *mut C_tensor;
14732 pub fn atg_split_with_sizes_copy(
14733 self_: *mut C_tensor,
14734 split_sizes_data: *const i64,
14735 split_sizes_len: c_int,
14736 dim_: i64,
14737 ) -> *mut *mut C_tensor;
14738 pub fn atg_split_with_sizes_copy_out(
14739 out_data: *const *mut C_tensor,
14740 out_len: c_int,
14741 self_: *mut C_tensor,
14742 split_sizes_data: *const i64,
14743 split_sizes_len: c_int,
14744 dim_: i64,
14745 );
14746 pub fn atg_sqrt(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14747 pub fn atg_sqrt_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14748 pub fn atg_sqrt_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
14749 pub fn atg_square(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14750 pub fn atg_square_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14751 pub fn atg_square_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
14752 pub fn atg_squeeze(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14753 pub fn atg_squeeze_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14754 pub fn atg_squeeze_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14755 pub fn atg_squeeze_copy_dim(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
14756 pub fn atg_squeeze_copy_dim_out(
14757 out__: *mut *mut C_tensor,
14758 out_: *mut C_tensor,
14759 self_: *mut C_tensor,
14760 dim_: i64,
14761 );
14762 pub fn atg_squeeze_copy_dims(
14763 out__: *mut *mut C_tensor,
14764 self_: *mut C_tensor,
14765 dim_data: *const i64,
14766 dim_len: c_int,
14767 );
14768 pub fn atg_squeeze_copy_dims_out(
14769 out__: *mut *mut C_tensor,
14770 out_: *mut C_tensor,
14771 self_: *mut C_tensor,
14772 dim_data: *const i64,
14773 dim_len: c_int,
14774 );
14775 pub fn atg_squeeze_copy_out(
14776 out__: *mut *mut C_tensor,
14777 out_: *mut C_tensor,
14778 self_: *mut C_tensor,
14779 );
14780 pub fn atg_squeeze_dim(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
14781 pub fn atg_squeeze_dim_(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
14782 pub fn atg_squeeze_dims(
14783 out__: *mut *mut C_tensor,
14784 self_: *mut C_tensor,
14785 dim_data: *const i64,
14786 dim_len: c_int,
14787 );
14788 pub fn atg_squeeze_dims_(
14789 out__: *mut *mut C_tensor,
14790 self_: *mut C_tensor,
14791 dim_data: *const i64,
14792 dim_len: c_int,
14793 );
14794 pub fn atg_sspaddmm(
14795 out__: *mut *mut C_tensor,
14796 self_: *mut C_tensor,
14797 mat1_: *mut C_tensor,
14798 mat2_: *mut C_tensor,
14799 );
14800 pub fn atg_sspaddmm_out(
14801 out__: *mut *mut C_tensor,
14802 out_: *mut C_tensor,
14803 self_: *mut C_tensor,
14804 mat1_: *mut C_tensor,
14805 mat2_: *mut C_tensor,
14806 );
14807 pub fn atg_stack(
14808 out__: *mut *mut C_tensor,
14809 tensors_data: *const *mut C_tensor,
14810 tensors_len: c_int,
14811 dim_: i64,
14812 );
14813 pub fn atg_stack_out(
14814 out__: *mut *mut C_tensor,
14815 out_: *mut C_tensor,
14816 tensors_data: *const *mut C_tensor,
14817 tensors_len: c_int,
14818 dim_: i64,
14819 );
14820 pub fn atg_std(out__: *mut *mut C_tensor, self_: *mut C_tensor, unbiased_: c_int);
14821 pub fn atg_std_correction(
14822 out__: *mut *mut C_tensor,
14823 self_: *mut C_tensor,
14824 dim_data: *const i64,
14825 dim_len: c_int,
14826 correction_: *mut C_scalar,
14827 keepdim_: c_int,
14828 );
14829 pub fn atg_std_correction_out(
14830 out__: *mut *mut C_tensor,
14831 out_: *mut C_tensor,
14832 self_: *mut C_tensor,
14833 dim_data: *const i64,
14834 dim_len: c_int,
14835 correction_: *mut C_scalar,
14836 keepdim_: c_int,
14837 );
14838 pub fn atg_std_dim(
14839 out__: *mut *mut C_tensor,
14840 self_: *mut C_tensor,
14841 dim_data: *const i64,
14842 dim_len: c_int,
14843 unbiased_: c_int,
14844 keepdim_: c_int,
14845 );
14846 pub fn atg_std_mean(out__: *mut *mut C_tensor, self_: *mut C_tensor, unbiased_: c_int);
14847 pub fn atg_std_mean_correction(
14848 out__: *mut *mut C_tensor,
14849 self_: *mut C_tensor,
14850 dim_data: *const i64,
14851 dim_len: c_int,
14852 correction_: *mut C_scalar,
14853 keepdim_: c_int,
14854 );
14855 pub fn atg_std_mean_correction_out(
14856 out__: *mut *mut C_tensor,
14857 out0_: *mut C_tensor,
14858 out1_: *mut C_tensor,
14859 self_: *mut C_tensor,
14860 dim_data: *const i64,
14861 dim_len: c_int,
14862 correction_: *mut C_scalar,
14863 keepdim_: c_int,
14864 );
14865 pub fn atg_std_mean_dim(
14866 out__: *mut *mut C_tensor,
14867 self_: *mut C_tensor,
14868 dim_data: *const i64,
14869 dim_len: c_int,
14870 unbiased_: c_int,
14871 keepdim_: c_int,
14872 );
14873 pub fn atg_std_out(
14874 out__: *mut *mut C_tensor,
14875 out_: *mut C_tensor,
14876 self_: *mut C_tensor,
14877 dim_data: *const i64,
14878 dim_len: c_int,
14879 unbiased_: c_int,
14880 keepdim_: c_int,
14881 );
14882 pub fn atg_stft(
14883 out__: *mut *mut C_tensor,
14884 self_: *mut C_tensor,
14885 n_fft_: i64,
14886 hop_length_v: i64,
14887 hop_length_null: i8,
14888 win_length_v: i64,
14889 win_length_null: i8,
14890 window_: *mut C_tensor,
14891 normalized_: c_int,
14892 onesided_: c_int,
14893 return_complex_: c_int,
14894 align_to_window_: c_int,
14895 );
14896 pub fn atg_stft_center(
14897 out__: *mut *mut C_tensor,
14898 self_: *mut C_tensor,
14899 n_fft_: i64,
14900 hop_length_v: i64,
14901 hop_length_null: i8,
14902 win_length_v: i64,
14903 win_length_null: i8,
14904 window_: *mut C_tensor,
14905 center_: c_int,
14906 pad_mode_ptr: *const u8,
14907 pad_mode_len: c_int,
14908 normalized_: c_int,
14909 onesided_: c_int,
14910 return_complex_: c_int,
14911 align_to_window_: c_int,
14912 );
14913 pub fn atg_storage_offset(self_: *mut C_tensor) -> i64;
14914 pub fn atg_sub(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
14915 pub fn atg_sub_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
14916 pub fn atg_sub_out(
14917 out__: *mut *mut C_tensor,
14918 out_: *mut C_tensor,
14919 self_: *mut C_tensor,
14920 other_: *mut C_tensor,
14921 );
14922 pub fn atg_sub_scalar(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
14923 pub fn atg_sub_scalar_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_scalar);
14924 pub fn atg_sub_scalar_out(
14925 out__: *mut *mut C_tensor,
14926 out_: *mut C_tensor,
14927 self_: *mut C_tensor,
14928 other_: *mut C_scalar,
14929 );
14930 pub fn atg_subtract(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
14931 pub fn atg_subtract_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
14932 pub fn atg_subtract_out(
14933 out__: *mut *mut C_tensor,
14934 out_: *mut C_tensor,
14935 self_: *mut C_tensor,
14936 other_: *mut C_tensor,
14937 );
14938 pub fn atg_subtract_scalar(
14939 out__: *mut *mut C_tensor,
14940 self_: *mut C_tensor,
14941 other_: *mut C_scalar,
14942 );
14943 pub fn atg_subtract_scalar_(
14944 out__: *mut *mut C_tensor,
14945 self_: *mut C_tensor,
14946 other_: *mut C_scalar,
14947 );
14948 pub fn atg_sum(out__: *mut *mut C_tensor, self_: *mut C_tensor, dtype_: c_int);
14949 pub fn atg_sum_dim_intlist(
14950 out__: *mut *mut C_tensor,
14951 self_: *mut C_tensor,
14952 dim_data: *const i64,
14953 dim_len: c_int,
14954 keepdim_: c_int,
14955 dtype_: c_int,
14956 );
14957 pub fn atg_sum_intlist_out(
14958 out__: *mut *mut C_tensor,
14959 out_: *mut C_tensor,
14960 self_: *mut C_tensor,
14961 dim_data: *const i64,
14962 dim_len: c_int,
14963 keepdim_: c_int,
14964 dtype_: c_int,
14965 );
14966 pub fn atg_sum_out(
14967 out__: *mut *mut C_tensor,
14968 out_: *mut C_tensor,
14969 self_: *mut C_tensor,
14970 dtype_: c_int,
14971 );
14972 pub fn atg_sum_to_size(
14973 out__: *mut *mut C_tensor,
14974 self_: *mut C_tensor,
14975 size_data: *const i64,
14976 size_len: c_int,
14977 );
14978 pub fn atg_svd(
14979 out__: *mut *mut C_tensor,
14980 self_: *mut C_tensor,
14981 some_: c_int,
14982 compute_uv_: c_int,
14983 );
14984 pub fn atg_svd_u(
14985 out__: *mut *mut C_tensor,
14986 U_: *mut C_tensor,
14987 S_: *mut C_tensor,
14988 V_: *mut C_tensor,
14989 self_: *mut C_tensor,
14990 some_: c_int,
14991 compute_uv_: c_int,
14992 );
14993 pub fn atg_swapaxes(out__: *mut *mut C_tensor, self_: *mut C_tensor, axis0_: i64, axis1_: i64);
14994 pub fn atg_swapaxes_(out__: *mut *mut C_tensor, self_: *mut C_tensor, axis0_: i64, axis1_: i64);
14995 pub fn atg_swapdims(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim0_: i64, dim1_: i64);
14996 pub fn atg_swapdims_(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim0_: i64, dim1_: i64);
14997 pub fn atg_t(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14998 pub fn atg_t_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
14999 pub fn atg_t_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15000 pub fn atg_t_copy_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
15001 pub fn atg_take(out__: *mut *mut C_tensor, self_: *mut C_tensor, index_: *mut C_tensor);
15002 pub fn atg_take_along_dim(
15003 out__: *mut *mut C_tensor,
15004 self_: *mut C_tensor,
15005 indices_: *mut C_tensor,
15006 dim_v: i64,
15007 dim_null: i8,
15008 );
15009 pub fn atg_take_along_dim_out(
15010 out__: *mut *mut C_tensor,
15011 out_: *mut C_tensor,
15012 self_: *mut C_tensor,
15013 indices_: *mut C_tensor,
15014 dim_v: i64,
15015 dim_null: i8,
15016 );
15017 pub fn atg_take_out(
15018 out__: *mut *mut C_tensor,
15019 out_: *mut C_tensor,
15020 self_: *mut C_tensor,
15021 index_: *mut C_tensor,
15022 );
15023 pub fn atg_tan(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15024 pub fn atg_tan_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15025 pub fn atg_tan_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
15026 pub fn atg_tanh(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15027 pub fn atg_tanh_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15028 pub fn atg_tanh_backward(
15029 out__: *mut *mut C_tensor,
15030 grad_output_: *mut C_tensor,
15031 output_: *mut C_tensor,
15032 );
15033 pub fn atg_tanh_backward_grad_input(
15034 out__: *mut *mut C_tensor,
15035 grad_input_: *mut C_tensor,
15036 grad_output_: *mut C_tensor,
15037 output_: *mut C_tensor,
15038 );
15039 pub fn atg_tanh_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
15040 pub fn atg_tensor_split(self_: *mut C_tensor, sections_: i64, dim_: i64) -> *mut *mut C_tensor;
15041 pub fn atg_tensor_split_indices(
15042 self_: *mut C_tensor,
15043 indices_data: *const i64,
15044 indices_len: c_int,
15045 dim_: i64,
15046 ) -> *mut *mut C_tensor;
15047 pub fn atg_tensor_split_tensor_indices_or_sections(
15048 self_: *mut C_tensor,
15049 tensor_indices_or_sections_: *mut C_tensor,
15050 dim_: i64,
15051 ) -> *mut *mut C_tensor;
15052 pub fn atg_tensordot(
15053 out__: *mut *mut C_tensor,
15054 self_: *mut C_tensor,
15055 other_: *mut C_tensor,
15056 dims_self_data: *const i64,
15057 dims_self_len: c_int,
15058 dims_other_data: *const i64,
15059 dims_other_len: c_int,
15060 );
15061 pub fn atg_tensordot_out(
15062 out__: *mut *mut C_tensor,
15063 out_: *mut C_tensor,
15064 self_: *mut C_tensor,
15065 other_: *mut C_tensor,
15066 dims_self_data: *const i64,
15067 dims_self_len: c_int,
15068 dims_other_data: *const i64,
15069 dims_other_len: c_int,
15070 );
15071 pub fn atg_threshold(
15072 out__: *mut *mut C_tensor,
15073 self_: *mut C_tensor,
15074 threshold_: *mut C_scalar,
15075 value_: *mut C_scalar,
15076 );
15077 pub fn atg_threshold_(
15078 out__: *mut *mut C_tensor,
15079 self_: *mut C_tensor,
15080 threshold_: *mut C_scalar,
15081 value_: *mut C_scalar,
15082 );
15083 pub fn atg_threshold_backward(
15084 out__: *mut *mut C_tensor,
15085 grad_output_: *mut C_tensor,
15086 self_: *mut C_tensor,
15087 threshold_: *mut C_scalar,
15088 );
15089 pub fn atg_threshold_backward_grad_input(
15090 out__: *mut *mut C_tensor,
15091 grad_input_: *mut C_tensor,
15092 grad_output_: *mut C_tensor,
15093 self_: *mut C_tensor,
15094 threshold_: *mut C_scalar,
15095 );
15096 pub fn atg_threshold_out(
15097 out__: *mut *mut C_tensor,
15098 out_: *mut C_tensor,
15099 self_: *mut C_tensor,
15100 threshold_: *mut C_scalar,
15101 value_: *mut C_scalar,
15102 );
15103 pub fn atg_tile(
15104 out__: *mut *mut C_tensor,
15105 self_: *mut C_tensor,
15106 dims_data: *const i64,
15107 dims_len: c_int,
15108 );
15109 pub fn atg_to(out__: *mut *mut C_tensor, self_: *mut C_tensor, device_: c_int);
15110 pub fn atg_to_dense(
15111 out__: *mut *mut C_tensor,
15112 self_: *mut C_tensor,
15113 dtype_: c_int,
15114 masked_grad_: c_int,
15115 );
15116 pub fn atg_to_dense_backward(
15117 out__: *mut *mut C_tensor,
15118 grad_: *mut C_tensor,
15119 input_: *mut C_tensor,
15120 masked_grad_: c_int,
15121 );
15122 pub fn atg_to_device(
15123 out__: *mut *mut C_tensor,
15124 self_: *mut C_tensor,
15125 device_: c_int,
15126 dtype_: c_int,
15127 non_blocking_: c_int,
15128 copy_: c_int,
15129 );
15130 pub fn atg_to_dtype(
15131 out__: *mut *mut C_tensor,
15132 self_: *mut C_tensor,
15133 dtype_: c_int,
15134 non_blocking_: c_int,
15135 copy_: c_int,
15136 );
15137 pub fn atg_to_dtype_layout(
15138 out__: *mut *mut C_tensor,
15139 self_: *mut C_tensor,
15140 options_kind: c_int,
15141 options_device: c_int,
15142 non_blocking_: c_int,
15143 copy_: c_int,
15144 );
15145 pub fn atg_to_mkldnn(out__: *mut *mut C_tensor, self_: *mut C_tensor, dtype_: c_int);
15146 pub fn atg_to_mkldnn_backward(
15147 out__: *mut *mut C_tensor,
15148 grad_: *mut C_tensor,
15149 input_: *mut C_tensor,
15150 );
15151 pub fn atg_to_mkldnn_out(
15152 out__: *mut *mut C_tensor,
15153 out_: *mut C_tensor,
15154 self_: *mut C_tensor,
15155 dtype_: c_int,
15156 );
15157 pub fn atg_to_other(
15158 out__: *mut *mut C_tensor,
15159 self_: *mut C_tensor,
15160 other_: *mut C_tensor,
15161 non_blocking_: c_int,
15162 copy_: c_int,
15163 );
15164 pub fn atg_to_padded_tensor(
15165 out__: *mut *mut C_tensor,
15166 self_: *mut C_tensor,
15167 padding_: f64,
15168 output_size_data: *const i64,
15169 output_size_len: c_int,
15170 );
15171 pub fn atg_to_padded_tensor_out(
15172 out__: *mut *mut C_tensor,
15173 out_: *mut C_tensor,
15174 self_: *mut C_tensor,
15175 padding_: f64,
15176 output_size_data: *const i64,
15177 output_size_len: c_int,
15178 );
15179 pub fn atg_to_sparse(
15180 out__: *mut *mut C_tensor,
15181 self_: *mut C_tensor,
15182 layout_: i8,
15183 blocksize_data: *const i64,
15184 blocksize_len: c_int,
15185 dense_dim_v: i64,
15186 dense_dim_null: i8,
15187 );
15188 pub fn atg_to_sparse_bsc(
15189 out__: *mut *mut C_tensor,
15190 self_: *mut C_tensor,
15191 blocksize_data: *const i64,
15192 blocksize_len: c_int,
15193 dense_dim_v: i64,
15194 dense_dim_null: i8,
15195 );
15196 pub fn atg_to_sparse_bsr(
15197 out__: *mut *mut C_tensor,
15198 self_: *mut C_tensor,
15199 blocksize_data: *const i64,
15200 blocksize_len: c_int,
15201 dense_dim_v: i64,
15202 dense_dim_null: i8,
15203 );
15204 pub fn atg_to_sparse_csc(
15205 out__: *mut *mut C_tensor,
15206 self_: *mut C_tensor,
15207 dense_dim_v: i64,
15208 dense_dim_null: i8,
15209 );
15210 pub fn atg_to_sparse_csr(
15211 out__: *mut *mut C_tensor,
15212 self_: *mut C_tensor,
15213 dense_dim_v: i64,
15214 dense_dim_null: i8,
15215 );
15216 pub fn atg_to_sparse_sparse_dim(
15217 out__: *mut *mut C_tensor,
15218 self_: *mut C_tensor,
15219 sparse_dim_: i64,
15220 );
15221 pub fn atg_topk(
15222 out__: *mut *mut C_tensor,
15223 self_: *mut C_tensor,
15224 k_: i64,
15225 dim_: i64,
15226 largest_: c_int,
15227 sorted_: c_int,
15228 );
15229 pub fn atg_topk_values(
15230 out__: *mut *mut C_tensor,
15231 values_: *mut C_tensor,
15232 indices_: *mut C_tensor,
15233 self_: *mut C_tensor,
15234 k_: i64,
15235 dim_: i64,
15236 largest_: c_int,
15237 sorted_: c_int,
15238 );
15239 pub fn atg_totype(out__: *mut *mut C_tensor, self_: *mut C_tensor, scalar_type_: c_int);
15240 pub fn atg_trace(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15241 pub fn atg_trace_backward(
15242 out__: *mut *mut C_tensor,
15243 grad_: *mut C_tensor,
15244 sizes_data: *const i64,
15245 sizes_len: c_int,
15246 );
15247 pub fn atg_trace_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
15248 pub fn atg_transpose(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim0_: i64, dim1_: i64);
15249 pub fn atg_transpose_(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim0_: i64, dim1_: i64);
15250 pub fn atg_transpose_copy(
15251 out__: *mut *mut C_tensor,
15252 self_: *mut C_tensor,
15253 dim0_: i64,
15254 dim1_: i64,
15255 );
15256 pub fn atg_transpose_copy_int_out(
15257 out__: *mut *mut C_tensor,
15258 out_: *mut C_tensor,
15259 self_: *mut C_tensor,
15260 dim0_: i64,
15261 dim1_: i64,
15262 );
15263 pub fn atg_trapezoid(out__: *mut *mut C_tensor, y_: *mut C_tensor, dim_: i64);
15264 pub fn atg_trapezoid_x(
15265 out__: *mut *mut C_tensor,
15266 y_: *mut C_tensor,
15267 x_: *mut C_tensor,
15268 dim_: i64,
15269 );
15270 pub fn atg_trapz(out__: *mut *mut C_tensor, y_: *mut C_tensor, x_: *mut C_tensor, dim_: i64);
15271 pub fn atg_trapz_dx(out__: *mut *mut C_tensor, y_: *mut C_tensor, dx_: f64, dim_: i64);
15272 pub fn atg_triangular_solve(
15273 out__: *mut *mut C_tensor,
15274 self_: *mut C_tensor,
15275 A_: *mut C_tensor,
15276 upper_: c_int,
15277 transpose_: c_int,
15278 unitriangular_: c_int,
15279 );
15280 pub fn atg_triangular_solve_x(
15281 out__: *mut *mut C_tensor,
15282 X_: *mut C_tensor,
15283 M_: *mut C_tensor,
15284 self_: *mut C_tensor,
15285 A_: *mut C_tensor,
15286 upper_: c_int,
15287 transpose_: c_int,
15288 unitriangular_: c_int,
15289 );
15290 pub fn atg_tril(out__: *mut *mut C_tensor, self_: *mut C_tensor, diagonal_: i64);
15291 pub fn atg_tril_(out__: *mut *mut C_tensor, self_: *mut C_tensor, diagonal_: i64);
15292 pub fn atg_tril_indices(
15293 out__: *mut *mut C_tensor,
15294 row_: i64,
15295 col_: i64,
15296 offset_: i64,
15297 options_kind: c_int,
15298 options_device: c_int,
15299 );
15300 pub fn atg_tril_indices_out(
15301 out__: *mut *mut C_tensor,
15302 out_: *mut C_tensor,
15303 row_: i64,
15304 col_: i64,
15305 offset_: i64,
15306 );
15307 pub fn atg_tril_out(
15308 out__: *mut *mut C_tensor,
15309 out_: *mut C_tensor,
15310 self_: *mut C_tensor,
15311 diagonal_: i64,
15312 );
15313 pub fn atg_triplet_margin_loss(
15314 out__: *mut *mut C_tensor,
15315 anchor_: *mut C_tensor,
15316 positive_: *mut C_tensor,
15317 negative_: *mut C_tensor,
15318 margin_: f64,
15319 p_: f64,
15320 eps_: f64,
15321 swap_: c_int,
15322 reduction_: i64,
15323 );
15324 pub fn atg_triu(out__: *mut *mut C_tensor, self_: *mut C_tensor, diagonal_: i64);
15325 pub fn atg_triu_(out__: *mut *mut C_tensor, self_: *mut C_tensor, diagonal_: i64);
15326 pub fn atg_triu_indices(
15327 out__: *mut *mut C_tensor,
15328 row_: i64,
15329 col_: i64,
15330 offset_: i64,
15331 options_kind: c_int,
15332 options_device: c_int,
15333 );
15334 pub fn atg_triu_indices_out(
15335 out__: *mut *mut C_tensor,
15336 out_: *mut C_tensor,
15337 row_: i64,
15338 col_: i64,
15339 offset_: i64,
15340 );
15341 pub fn atg_triu_out(
15342 out__: *mut *mut C_tensor,
15343 out_: *mut C_tensor,
15344 self_: *mut C_tensor,
15345 diagonal_: i64,
15346 );
15347 pub fn atg_true_divide(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
15348 pub fn atg_true_divide_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
15349 pub fn atg_true_divide_out(
15350 out__: *mut *mut C_tensor,
15351 out_: *mut C_tensor,
15352 self_: *mut C_tensor,
15353 other_: *mut C_tensor,
15354 );
15355 pub fn atg_true_divide_scalar(
15356 out__: *mut *mut C_tensor,
15357 self_: *mut C_tensor,
15358 other_: *mut C_scalar,
15359 );
15360 pub fn atg_true_divide_scalar_(
15361 out__: *mut *mut C_tensor,
15362 self_: *mut C_tensor,
15363 other_: *mut C_scalar,
15364 );
15365 pub fn atg_trunc(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15366 pub fn atg_trunc_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15367 pub fn atg_trunc_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
15368 pub fn atg_type_as(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
15369 pub fn atg_unbind(self_: *mut C_tensor, dim_: i64) -> *mut *mut C_tensor;
15370 pub fn atg_unbind_copy(self_: *mut C_tensor, dim_: i64) -> *mut *mut C_tensor;
15371 pub fn atg_unbind_copy_int_out(
15372 out_data: *const *mut C_tensor,
15373 out_len: c_int,
15374 self_: *mut C_tensor,
15375 dim_: i64,
15376 );
15377 pub fn atg_unflatten(
15378 out__: *mut *mut C_tensor,
15379 self_: *mut C_tensor,
15380 dim_: i64,
15381 sizes_data: *const i64,
15382 sizes_len: c_int,
15383 );
15384 pub fn atg_unflatten_dense_tensors(
15385 flat_: *mut C_tensor,
15386 tensors_data: *const *mut C_tensor,
15387 tensors_len: c_int,
15388 ) -> *mut *mut C_tensor;
15389 pub fn atg_unfold(
15390 out__: *mut *mut C_tensor,
15391 self_: *mut C_tensor,
15392 dimension_: i64,
15393 size_: i64,
15394 step_: i64,
15395 );
15396 pub fn atg_unfold_backward(
15397 out__: *mut *mut C_tensor,
15398 grad_in_: *mut C_tensor,
15399 input_sizes_data: *const i64,
15400 input_sizes_len: c_int,
15401 dim_: i64,
15402 size_: i64,
15403 step_: i64,
15404 );
15405 pub fn atg_unfold_backward_out(
15406 out__: *mut *mut C_tensor,
15407 out_: *mut C_tensor,
15408 grad_in_: *mut C_tensor,
15409 input_sizes_data: *const i64,
15410 input_sizes_len: c_int,
15411 dim_: i64,
15412 size_: i64,
15413 step_: i64,
15414 );
15415 pub fn atg_unfold_copy(
15416 out__: *mut *mut C_tensor,
15417 self_: *mut C_tensor,
15418 dimension_: i64,
15419 size_: i64,
15420 step_: i64,
15421 );
15422 pub fn atg_unfold_copy_out(
15423 out__: *mut *mut C_tensor,
15424 out_: *mut C_tensor,
15425 self_: *mut C_tensor,
15426 dimension_: i64,
15427 size_: i64,
15428 step_: i64,
15429 );
15430 pub fn atg_uniform(out__: *mut *mut C_tensor, self_: *mut C_tensor, from_: f64, to_: f64);
15431 pub fn atg_uniform_(out__: *mut *mut C_tensor, self_: *mut C_tensor, from_: f64, to_: f64);
15432 pub fn atg_uniform_out(
15433 out__: *mut *mut C_tensor,
15434 out_: *mut C_tensor,
15435 self_: *mut C_tensor,
15436 from_: f64,
15437 to_: f64,
15438 );
15439 pub fn atg_unique_consecutive(
15440 out__: *mut *mut C_tensor,
15441 self_: *mut C_tensor,
15442 return_inverse_: c_int,
15443 return_counts_: c_int,
15444 dim_v: i64,
15445 dim_null: i8,
15446 );
15447 pub fn atg_unique_consecutive_out(
15448 out__: *mut *mut C_tensor,
15449 out0_: *mut C_tensor,
15450 out1_: *mut C_tensor,
15451 out2_: *mut C_tensor,
15452 self_: *mut C_tensor,
15453 return_inverse_: c_int,
15454 return_counts_: c_int,
15455 dim_v: i64,
15456 dim_null: i8,
15457 );
15458 pub fn atg_unique_dim(
15459 out__: *mut *mut C_tensor,
15460 self_: *mut C_tensor,
15461 dim_: i64,
15462 sorted_: c_int,
15463 return_inverse_: c_int,
15464 return_counts_: c_int,
15465 );
15466 pub fn atg_unique_dim_consecutive(
15467 out__: *mut *mut C_tensor,
15468 self_: *mut C_tensor,
15469 dim_: i64,
15470 return_inverse_: c_int,
15471 return_counts_: c_int,
15472 );
15473 pub fn atg_unique_dim_consecutive_out(
15474 out__: *mut *mut C_tensor,
15475 out0_: *mut C_tensor,
15476 out1_: *mut C_tensor,
15477 out2_: *mut C_tensor,
15478 self_: *mut C_tensor,
15479 dim_: i64,
15480 return_inverse_: c_int,
15481 return_counts_: c_int,
15482 );
15483 pub fn atg_unique_dim_out(
15484 out__: *mut *mut C_tensor,
15485 out0_: *mut C_tensor,
15486 out1_: *mut C_tensor,
15487 out2_: *mut C_tensor,
15488 self_: *mut C_tensor,
15489 dim_: i64,
15490 sorted_: c_int,
15491 return_inverse_: c_int,
15492 return_counts_: c_int,
15493 );
15494 pub fn atg_unsafe_chunk(self_: *mut C_tensor, chunks_: i64, dim_: i64) -> *mut *mut C_tensor;
15495 pub fn atg_unsafe_split(
15496 self_: *mut C_tensor,
15497 split_size_: i64,
15498 dim_: i64,
15499 ) -> *mut *mut C_tensor;
15500 pub fn atg_unsafe_split_tensor_out(
15501 out_data: *const *mut C_tensor,
15502 out_len: c_int,
15503 self_: *mut C_tensor,
15504 split_size_: i64,
15505 dim_: i64,
15506 );
15507 pub fn atg_unsafe_split_with_sizes(
15508 self_: *mut C_tensor,
15509 split_sizes_data: *const i64,
15510 split_sizes_len: c_int,
15511 dim_: i64,
15512 ) -> *mut *mut C_tensor;
15513 pub fn atg_unsafe_split_with_sizes_out(
15514 out_data: *const *mut C_tensor,
15515 out_len: c_int,
15516 self_: *mut C_tensor,
15517 split_sizes_data: *const i64,
15518 split_sizes_len: c_int,
15519 dim_: i64,
15520 );
15521 pub fn atg_unsqueeze(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
15522 pub fn atg_unsqueeze_(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
15523 pub fn atg_unsqueeze_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor, dim_: i64);
15524 pub fn atg_unsqueeze_copy_out(
15525 out__: *mut *mut C_tensor,
15526 out_: *mut C_tensor,
15527 self_: *mut C_tensor,
15528 dim_: i64,
15529 );
15530 pub fn atg_upsample_bicubic2d(
15531 out__: *mut *mut C_tensor,
15532 self_: *mut C_tensor,
15533 output_size_data: *const i64,
15534 output_size_len: c_int,
15535 align_corners_: c_int,
15536 scales_h_v: f64,
15537 scales_h_null: i8,
15538 scales_w_v: f64,
15539 scales_w_null: i8,
15540 );
15541 pub fn atg_upsample_bicubic2d_backward(
15542 out__: *mut *mut C_tensor,
15543 grad_output_: *mut C_tensor,
15544 output_size_data: *const i64,
15545 output_size_len: c_int,
15546 input_size_data: *const i64,
15547 input_size_len: c_int,
15548 align_corners_: c_int,
15549 scales_h_v: f64,
15550 scales_h_null: i8,
15551 scales_w_v: f64,
15552 scales_w_null: i8,
15553 );
15554 pub fn atg_upsample_bicubic2d_backward_grad_input(
15555 out__: *mut *mut C_tensor,
15556 grad_input_: *mut C_tensor,
15557 grad_output_: *mut C_tensor,
15558 output_size_data: *const i64,
15559 output_size_len: c_int,
15560 input_size_data: *const i64,
15561 input_size_len: c_int,
15562 align_corners_: c_int,
15563 scales_h_v: f64,
15564 scales_h_null: i8,
15565 scales_w_v: f64,
15566 scales_w_null: i8,
15567 );
15568 pub fn atg_upsample_bicubic2d_out(
15569 out__: *mut *mut C_tensor,
15570 out_: *mut C_tensor,
15571 self_: *mut C_tensor,
15572 output_size_data: *const i64,
15573 output_size_len: c_int,
15574 align_corners_: c_int,
15575 scales_h_v: f64,
15576 scales_h_null: i8,
15577 scales_w_v: f64,
15578 scales_w_null: i8,
15579 );
15580 pub fn atg_upsample_bicubic2d_vec(
15581 out__: *mut *mut C_tensor,
15582 input_: *mut C_tensor,
15583 output_size_data: *const i64,
15584 output_size_len: c_int,
15585 align_corners_: c_int,
15586 scale_factors_data: *const f64,
15587 scale_factors_len: c_int,
15588 );
15589 pub fn atg_upsample_bilinear2d(
15590 out__: *mut *mut C_tensor,
15591 self_: *mut C_tensor,
15592 output_size_data: *const i64,
15593 output_size_len: c_int,
15594 align_corners_: c_int,
15595 scales_h_v: f64,
15596 scales_h_null: i8,
15597 scales_w_v: f64,
15598 scales_w_null: i8,
15599 );
15600 pub fn atg_upsample_bilinear2d_backward(
15601 out__: *mut *mut C_tensor,
15602 grad_output_: *mut C_tensor,
15603 output_size_data: *const i64,
15604 output_size_len: c_int,
15605 input_size_data: *const i64,
15606 input_size_len: c_int,
15607 align_corners_: c_int,
15608 scales_h_v: f64,
15609 scales_h_null: i8,
15610 scales_w_v: f64,
15611 scales_w_null: i8,
15612 );
15613 pub fn atg_upsample_bilinear2d_backward_grad_input(
15614 out__: *mut *mut C_tensor,
15615 grad_input_: *mut C_tensor,
15616 grad_output_: *mut C_tensor,
15617 output_size_data: *const i64,
15618 output_size_len: c_int,
15619 input_size_data: *const i64,
15620 input_size_len: c_int,
15621 align_corners_: c_int,
15622 scales_h_v: f64,
15623 scales_h_null: i8,
15624 scales_w_v: f64,
15625 scales_w_null: i8,
15626 );
15627 pub fn atg_upsample_bilinear2d_out(
15628 out__: *mut *mut C_tensor,
15629 out_: *mut C_tensor,
15630 self_: *mut C_tensor,
15631 output_size_data: *const i64,
15632 output_size_len: c_int,
15633 align_corners_: c_int,
15634 scales_h_v: f64,
15635 scales_h_null: i8,
15636 scales_w_v: f64,
15637 scales_w_null: i8,
15638 );
15639 pub fn atg_upsample_bilinear2d_vec(
15640 out__: *mut *mut C_tensor,
15641 input_: *mut C_tensor,
15642 output_size_data: *const i64,
15643 output_size_len: c_int,
15644 align_corners_: c_int,
15645 scale_factors_data: *const f64,
15646 scale_factors_len: c_int,
15647 );
15648 pub fn atg_upsample_bilinear2d_vec_out(
15649 out__: *mut *mut C_tensor,
15650 out_: *mut C_tensor,
15651 input_: *mut C_tensor,
15652 output_size_data: *const i64,
15653 output_size_len: c_int,
15654 align_corners_: c_int,
15655 scale_factors_data: *const f64,
15656 scale_factors_len: c_int,
15657 );
15658 pub fn atg_upsample_linear1d(
15659 out__: *mut *mut C_tensor,
15660 self_: *mut C_tensor,
15661 output_size_data: *const i64,
15662 output_size_len: c_int,
15663 align_corners_: c_int,
15664 scales_v: f64,
15665 scales_null: i8,
15666 );
15667 pub fn atg_upsample_linear1d_backward(
15668 out__: *mut *mut C_tensor,
15669 grad_output_: *mut C_tensor,
15670 output_size_data: *const i64,
15671 output_size_len: c_int,
15672 input_size_data: *const i64,
15673 input_size_len: c_int,
15674 align_corners_: c_int,
15675 scales_v: f64,
15676 scales_null: i8,
15677 );
15678 pub fn atg_upsample_linear1d_backward_grad_input(
15679 out__: *mut *mut C_tensor,
15680 grad_input_: *mut C_tensor,
15681 grad_output_: *mut C_tensor,
15682 output_size_data: *const i64,
15683 output_size_len: c_int,
15684 input_size_data: *const i64,
15685 input_size_len: c_int,
15686 align_corners_: c_int,
15687 scales_v: f64,
15688 scales_null: i8,
15689 );
15690 pub fn atg_upsample_linear1d_out(
15691 out__: *mut *mut C_tensor,
15692 out_: *mut C_tensor,
15693 self_: *mut C_tensor,
15694 output_size_data: *const i64,
15695 output_size_len: c_int,
15696 align_corners_: c_int,
15697 scales_v: f64,
15698 scales_null: i8,
15699 );
15700 pub fn atg_upsample_linear1d_vec(
15701 out__: *mut *mut C_tensor,
15702 input_: *mut C_tensor,
15703 output_size_data: *const i64,
15704 output_size_len: c_int,
15705 align_corners_: c_int,
15706 scale_factors_data: *const f64,
15707 scale_factors_len: c_int,
15708 );
15709 pub fn atg_upsample_nearest1d(
15710 out__: *mut *mut C_tensor,
15711 self_: *mut C_tensor,
15712 output_size_data: *const i64,
15713 output_size_len: c_int,
15714 scales_v: f64,
15715 scales_null: i8,
15716 );
15717 pub fn atg_upsample_nearest1d_backward(
15718 out__: *mut *mut C_tensor,
15719 grad_output_: *mut C_tensor,
15720 output_size_data: *const i64,
15721 output_size_len: c_int,
15722 input_size_data: *const i64,
15723 input_size_len: c_int,
15724 scales_v: f64,
15725 scales_null: i8,
15726 );
15727 pub fn atg_upsample_nearest1d_backward_grad_input(
15728 out__: *mut *mut C_tensor,
15729 grad_input_: *mut C_tensor,
15730 grad_output_: *mut C_tensor,
15731 output_size_data: *const i64,
15732 output_size_len: c_int,
15733 input_size_data: *const i64,
15734 input_size_len: c_int,
15735 scales_v: f64,
15736 scales_null: i8,
15737 );
15738 pub fn atg_upsample_nearest1d_out(
15739 out__: *mut *mut C_tensor,
15740 out_: *mut C_tensor,
15741 self_: *mut C_tensor,
15742 output_size_data: *const i64,
15743 output_size_len: c_int,
15744 scales_v: f64,
15745 scales_null: i8,
15746 );
15747 pub fn atg_upsample_nearest1d_vec(
15748 out__: *mut *mut C_tensor,
15749 input_: *mut C_tensor,
15750 output_size_data: *const i64,
15751 output_size_len: c_int,
15752 scale_factors_data: *const f64,
15753 scale_factors_len: c_int,
15754 );
15755 pub fn atg_upsample_nearest2d(
15756 out__: *mut *mut C_tensor,
15757 self_: *mut C_tensor,
15758 output_size_data: *const i64,
15759 output_size_len: c_int,
15760 scales_h_v: f64,
15761 scales_h_null: i8,
15762 scales_w_v: f64,
15763 scales_w_null: i8,
15764 );
15765 pub fn atg_upsample_nearest2d_backward(
15766 out__: *mut *mut C_tensor,
15767 grad_output_: *mut C_tensor,
15768 output_size_data: *const i64,
15769 output_size_len: c_int,
15770 input_size_data: *const i64,
15771 input_size_len: c_int,
15772 scales_h_v: f64,
15773 scales_h_null: i8,
15774 scales_w_v: f64,
15775 scales_w_null: i8,
15776 );
15777 pub fn atg_upsample_nearest2d_backward_grad_input(
15778 out__: *mut *mut C_tensor,
15779 grad_input_: *mut C_tensor,
15780 grad_output_: *mut C_tensor,
15781 output_size_data: *const i64,
15782 output_size_len: c_int,
15783 input_size_data: *const i64,
15784 input_size_len: c_int,
15785 scales_h_v: f64,
15786 scales_h_null: i8,
15787 scales_w_v: f64,
15788 scales_w_null: i8,
15789 );
15790 pub fn atg_upsample_nearest2d_out(
15791 out__: *mut *mut C_tensor,
15792 out_: *mut C_tensor,
15793 self_: *mut C_tensor,
15794 output_size_data: *const i64,
15795 output_size_len: c_int,
15796 scales_h_v: f64,
15797 scales_h_null: i8,
15798 scales_w_v: f64,
15799 scales_w_null: i8,
15800 );
15801 pub fn atg_upsample_nearest2d_vec(
15802 out__: *mut *mut C_tensor,
15803 input_: *mut C_tensor,
15804 output_size_data: *const i64,
15805 output_size_len: c_int,
15806 scale_factors_data: *const f64,
15807 scale_factors_len: c_int,
15808 );
15809 pub fn atg_upsample_nearest2d_vec_out(
15810 out__: *mut *mut C_tensor,
15811 out_: *mut C_tensor,
15812 input_: *mut C_tensor,
15813 output_size_data: *const i64,
15814 output_size_len: c_int,
15815 scale_factors_data: *const f64,
15816 scale_factors_len: c_int,
15817 );
15818 pub fn atg_upsample_nearest3d(
15819 out__: *mut *mut C_tensor,
15820 self_: *mut C_tensor,
15821 output_size_data: *const i64,
15822 output_size_len: c_int,
15823 scales_d_v: f64,
15824 scales_d_null: i8,
15825 scales_h_v: f64,
15826 scales_h_null: i8,
15827 scales_w_v: f64,
15828 scales_w_null: i8,
15829 );
15830 pub fn atg_upsample_nearest3d_backward(
15831 out__: *mut *mut C_tensor,
15832 grad_output_: *mut C_tensor,
15833 output_size_data: *const i64,
15834 output_size_len: c_int,
15835 input_size_data: *const i64,
15836 input_size_len: c_int,
15837 scales_d_v: f64,
15838 scales_d_null: i8,
15839 scales_h_v: f64,
15840 scales_h_null: i8,
15841 scales_w_v: f64,
15842 scales_w_null: i8,
15843 );
15844 pub fn atg_upsample_nearest3d_backward_grad_input(
15845 out__: *mut *mut C_tensor,
15846 grad_input_: *mut C_tensor,
15847 grad_output_: *mut C_tensor,
15848 output_size_data: *const i64,
15849 output_size_len: c_int,
15850 input_size_data: *const i64,
15851 input_size_len: c_int,
15852 scales_d_v: f64,
15853 scales_d_null: i8,
15854 scales_h_v: f64,
15855 scales_h_null: i8,
15856 scales_w_v: f64,
15857 scales_w_null: i8,
15858 );
15859 pub fn atg_upsample_nearest3d_out(
15860 out__: *mut *mut C_tensor,
15861 out_: *mut C_tensor,
15862 self_: *mut C_tensor,
15863 output_size_data: *const i64,
15864 output_size_len: c_int,
15865 scales_d_v: f64,
15866 scales_d_null: i8,
15867 scales_h_v: f64,
15868 scales_h_null: i8,
15869 scales_w_v: f64,
15870 scales_w_null: i8,
15871 );
15872 pub fn atg_upsample_nearest3d_vec(
15873 out__: *mut *mut C_tensor,
15874 input_: *mut C_tensor,
15875 output_size_data: *const i64,
15876 output_size_len: c_int,
15877 scale_factors_data: *const f64,
15878 scale_factors_len: c_int,
15879 );
15880 pub fn atg_upsample_trilinear3d(
15881 out__: *mut *mut C_tensor,
15882 self_: *mut C_tensor,
15883 output_size_data: *const i64,
15884 output_size_len: c_int,
15885 align_corners_: c_int,
15886 scales_d_v: f64,
15887 scales_d_null: i8,
15888 scales_h_v: f64,
15889 scales_h_null: i8,
15890 scales_w_v: f64,
15891 scales_w_null: i8,
15892 );
15893 pub fn atg_upsample_trilinear3d_backward(
15894 out__: *mut *mut C_tensor,
15895 grad_output_: *mut C_tensor,
15896 output_size_data: *const i64,
15897 output_size_len: c_int,
15898 input_size_data: *const i64,
15899 input_size_len: c_int,
15900 align_corners_: c_int,
15901 scales_d_v: f64,
15902 scales_d_null: i8,
15903 scales_h_v: f64,
15904 scales_h_null: i8,
15905 scales_w_v: f64,
15906 scales_w_null: i8,
15907 );
15908 pub fn atg_upsample_trilinear3d_backward_grad_input(
15909 out__: *mut *mut C_tensor,
15910 grad_input_: *mut C_tensor,
15911 grad_output_: *mut C_tensor,
15912 output_size_data: *const i64,
15913 output_size_len: c_int,
15914 input_size_data: *const i64,
15915 input_size_len: c_int,
15916 align_corners_: c_int,
15917 scales_d_v: f64,
15918 scales_d_null: i8,
15919 scales_h_v: f64,
15920 scales_h_null: i8,
15921 scales_w_v: f64,
15922 scales_w_null: i8,
15923 );
15924 pub fn atg_upsample_trilinear3d_out(
15925 out__: *mut *mut C_tensor,
15926 out_: *mut C_tensor,
15927 self_: *mut C_tensor,
15928 output_size_data: *const i64,
15929 output_size_len: c_int,
15930 align_corners_: c_int,
15931 scales_d_v: f64,
15932 scales_d_null: i8,
15933 scales_h_v: f64,
15934 scales_h_null: i8,
15935 scales_w_v: f64,
15936 scales_w_null: i8,
15937 );
15938 pub fn atg_upsample_trilinear3d_vec(
15939 out__: *mut *mut C_tensor,
15940 input_: *mut C_tensor,
15941 output_size_data: *const i64,
15942 output_size_len: c_int,
15943 align_corners_: c_int,
15944 scale_factors_data: *const f64,
15945 scale_factors_len: c_int,
15946 );
15947 pub fn atg_value_selecting_reduction_backward(
15948 out__: *mut *mut C_tensor,
15949 grad_: *mut C_tensor,
15950 dim_: i64,
15951 indices_: *mut C_tensor,
15952 sizes_data: *const i64,
15953 sizes_len: c_int,
15954 keepdim_: c_int,
15955 );
15956 pub fn atg_values(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15957 pub fn atg_values_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
15958 pub fn atg_values_copy_out(
15959 out__: *mut *mut C_tensor,
15960 out_: *mut C_tensor,
15961 self_: *mut C_tensor,
15962 );
15963 pub fn atg_vander(
15964 out__: *mut *mut C_tensor,
15965 x_: *mut C_tensor,
15966 n_v: i64,
15967 n_null: i8,
15968 increasing_: c_int,
15969 );
15970 pub fn atg_var(out__: *mut *mut C_tensor, self_: *mut C_tensor, unbiased_: c_int);
15971 pub fn atg_var_correction(
15972 out__: *mut *mut C_tensor,
15973 self_: *mut C_tensor,
15974 dim_data: *const i64,
15975 dim_len: c_int,
15976 correction_: *mut C_scalar,
15977 keepdim_: c_int,
15978 );
15979 pub fn atg_var_correction_out(
15980 out__: *mut *mut C_tensor,
15981 out_: *mut C_tensor,
15982 self_: *mut C_tensor,
15983 dim_data: *const i64,
15984 dim_len: c_int,
15985 correction_: *mut C_scalar,
15986 keepdim_: c_int,
15987 );
15988 pub fn atg_var_dim(
15989 out__: *mut *mut C_tensor,
15990 self_: *mut C_tensor,
15991 dim_data: *const i64,
15992 dim_len: c_int,
15993 unbiased_: c_int,
15994 keepdim_: c_int,
15995 );
15996 pub fn atg_var_mean(out__: *mut *mut C_tensor, self_: *mut C_tensor, unbiased_: c_int);
15997 pub fn atg_var_mean_correction(
15998 out__: *mut *mut C_tensor,
15999 self_: *mut C_tensor,
16000 dim_data: *const i64,
16001 dim_len: c_int,
16002 correction_: *mut C_scalar,
16003 keepdim_: c_int,
16004 );
16005 pub fn atg_var_mean_correction_out(
16006 out__: *mut *mut C_tensor,
16007 out0_: *mut C_tensor,
16008 out1_: *mut C_tensor,
16009 self_: *mut C_tensor,
16010 dim_data: *const i64,
16011 dim_len: c_int,
16012 correction_: *mut C_scalar,
16013 keepdim_: c_int,
16014 );
16015 pub fn atg_var_mean_dim(
16016 out__: *mut *mut C_tensor,
16017 self_: *mut C_tensor,
16018 dim_data: *const i64,
16019 dim_len: c_int,
16020 unbiased_: c_int,
16021 keepdim_: c_int,
16022 );
16023 pub fn atg_var_out(
16024 out__: *mut *mut C_tensor,
16025 out_: *mut C_tensor,
16026 self_: *mut C_tensor,
16027 dim_data: *const i64,
16028 dim_len: c_int,
16029 unbiased_: c_int,
16030 keepdim_: c_int,
16031 );
16032 pub fn atg_vdot(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
16033 pub fn atg_vdot_out(
16034 out__: *mut *mut C_tensor,
16035 out_: *mut C_tensor,
16036 self_: *mut C_tensor,
16037 other_: *mut C_tensor,
16038 );
16039 pub fn atg_view(
16040 out__: *mut *mut C_tensor,
16041 self_: *mut C_tensor,
16042 size_data: *const i64,
16043 size_len: c_int,
16044 );
16045 pub fn atg_view_as(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
16046 pub fn atg_view_as_complex(out__: *mut *mut C_tensor, self_: *mut C_tensor);
16047 pub fn atg_view_as_complex_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
16048 pub fn atg_view_as_complex_copy_out(
16049 out__: *mut *mut C_tensor,
16050 out_: *mut C_tensor,
16051 self_: *mut C_tensor,
16052 );
16053 pub fn atg_view_as_real(out__: *mut *mut C_tensor, self_: *mut C_tensor);
16054 pub fn atg_view_as_real_copy(out__: *mut *mut C_tensor, self_: *mut C_tensor);
16055 pub fn atg_view_as_real_copy_out(
16056 out__: *mut *mut C_tensor,
16057 out_: *mut C_tensor,
16058 self_: *mut C_tensor,
16059 );
16060 pub fn atg_view_copy(
16061 out__: *mut *mut C_tensor,
16062 self_: *mut C_tensor,
16063 size_data: *const i64,
16064 size_len: c_int,
16065 );
16066 pub fn atg_view_copy_dtype(out__: *mut *mut C_tensor, self_: *mut C_tensor, dtype_: c_int);
16067 pub fn atg_view_copy_dtype_out(
16068 out__: *mut *mut C_tensor,
16069 out_: *mut C_tensor,
16070 self_: *mut C_tensor,
16071 dtype_: c_int,
16072 );
16073 pub fn atg_view_copy_out(
16074 out__: *mut *mut C_tensor,
16075 out_: *mut C_tensor,
16076 self_: *mut C_tensor,
16077 size_data: *const i64,
16078 size_len: c_int,
16079 );
16080 pub fn atg_view_dtype(out__: *mut *mut C_tensor, self_: *mut C_tensor, dtype_: c_int);
16081 pub fn atg_vsplit(self_: *mut C_tensor, sections_: i64) -> *mut *mut C_tensor;
16082 pub fn atg_vsplit_array(
16083 self_: *mut C_tensor,
16084 indices_data: *const i64,
16085 indices_len: c_int,
16086 ) -> *mut *mut C_tensor;
16087 pub fn atg_vstack(
16088 out__: *mut *mut C_tensor,
16089 tensors_data: *const *mut C_tensor,
16090 tensors_len: c_int,
16091 );
16092 pub fn atg_vstack_out(
16093 out__: *mut *mut C_tensor,
16094 out_: *mut C_tensor,
16095 tensors_data: *const *mut C_tensor,
16096 tensors_len: c_int,
16097 );
16098 pub fn atg_where(condition_: *mut C_tensor) -> *mut *mut C_tensor;
16099 pub fn atg_where_scalar(
16100 out__: *mut *mut C_tensor,
16101 condition_: *mut C_tensor,
16102 self_scalar_: *mut C_scalar,
16103 other_: *mut C_scalar,
16104 );
16105 pub fn atg_where_scalarother(
16106 out__: *mut *mut C_tensor,
16107 condition_: *mut C_tensor,
16108 self_: *mut C_tensor,
16109 other_: *mut C_scalar,
16110 );
16111 pub fn atg_where_scalarself(
16112 out__: *mut *mut C_tensor,
16113 condition_: *mut C_tensor,
16114 self_scalar_: *mut C_scalar,
16115 other_: *mut C_tensor,
16116 );
16117 pub fn atg_where_self(
16118 out__: *mut *mut C_tensor,
16119 condition_: *mut C_tensor,
16120 self_: *mut C_tensor,
16121 other_: *mut C_tensor,
16122 );
16123 pub fn atg_where_self_out(
16124 out__: *mut *mut C_tensor,
16125 out_: *mut C_tensor,
16126 condition_: *mut C_tensor,
16127 self_: *mut C_tensor,
16128 other_: *mut C_tensor,
16129 );
16130 pub fn atg_xlogy(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
16131 pub fn atg_xlogy_(out__: *mut *mut C_tensor, self_: *mut C_tensor, other_: *mut C_tensor);
16132 pub fn atg_xlogy_outscalar_other(
16133 out__: *mut *mut C_tensor,
16134 out_: *mut C_tensor,
16135 self_: *mut C_tensor,
16136 other_: *mut C_scalar,
16137 );
16138 pub fn atg_xlogy_outscalar_self(
16139 out__: *mut *mut C_tensor,
16140 out_: *mut C_tensor,
16141 self_scalar_: *mut C_scalar,
16142 other_: *mut C_tensor,
16143 );
16144 pub fn atg_xlogy_outtensor(
16145 out__: *mut *mut C_tensor,
16146 out_: *mut C_tensor,
16147 self_: *mut C_tensor,
16148 other_: *mut C_tensor,
16149 );
16150 pub fn atg_xlogy_scalar_other(
16151 out__: *mut *mut C_tensor,
16152 self_: *mut C_tensor,
16153 other_: *mut C_scalar,
16154 );
16155 pub fn atg_xlogy_scalar_other_(
16156 out__: *mut *mut C_tensor,
16157 self_: *mut C_tensor,
16158 other_: *mut C_scalar,
16159 );
16160 pub fn atg_xlogy_scalar_self(
16161 out__: *mut *mut C_tensor,
16162 self_scalar_: *mut C_scalar,
16163 other_: *mut C_tensor,
16164 );
16165 pub fn atg_zero(out__: *mut *mut C_tensor, self_: *mut C_tensor);
16166 pub fn atg_zero_(out__: *mut *mut C_tensor, self_: *mut C_tensor);
16167 pub fn atg_zero_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
16168 pub fn atg_zeros(
16169 out__: *mut *mut C_tensor,
16170 size_data: *const i64,
16171 size_len: c_int,
16172 options_kind: c_int,
16173 options_device: c_int,
16174 );
16175 pub fn atg_zeros_like(out__: *mut *mut C_tensor, self_: *mut C_tensor);
16176 pub fn atg_zeros_like_out(out__: *mut *mut C_tensor, out_: *mut C_tensor, self_: *mut C_tensor);
16177 pub fn atg_zeros_out(
16178 out__: *mut *mut C_tensor,
16179 out_: *mut C_tensor,
16180 size_data: *const i64,
16181 size_len: c_int,
16182 );
16183}