Lines Matching full:at
3 // NB: Must be at the top of file to avoid including the deprecated "math.h".
51 const at::Tensor& t);
55 at::ArrayRef<at::Tensor> t);
56 TORCH_API at::Tensor copysign_tensor_self_backward(
60 TORCH_API at::Tensor not_implemented(const char* name, const char* reason = "");
64 at::Tensor handle_r_to_c(ScalarType self_st, Tensor gradient_result);
65 at::Tensor maybe_multiply(const at::Tensor& t, const at::Scalar& s);
75 at::Tensor norm_backward(
76 const at::Tensor& grad,
77 const at::Tensor& self,
78 const std::optional<at::Scalar>& p_,
79 const at::Tensor& norm);
80 at::Tensor norm_backward(
81 at::Tensor grad,
82 const at::Tensor& self,
83 const std::optional<at::Scalar>& p_,
84 at::Tensor norm,
85 at::IntArrayRef dim,
113 const at::OptionalIntArrayRef& opt_dim,
115 at::Tensor linalg_vector_norm_backward(
116 at::Tensor grad,
117 const at::Tensor& self,
118 const at::Scalar& ord,
119 at::Tensor norm,
120 const at::OptionalIntArrayRef& opt_dim,
122 at::Tensor pow_backward(
123 at::Tensor grad,
124 const at::Tensor& self,
125 const at::Scalar& exponent_);
126 at::Tensor pow_backward_self(
127 const at::Tensor& grad,
128 const at::Tensor& self,
129 const at::Tensor& exponent);
130 at::Tensor pow_backward_exponent(
131 const at::Tensor& grad,
132 const at::Tensor& self,
133 const at::Tensor& exponent,
134 const at::Tensor& result);
135 at::Tensor pow_backward_exponent(
136 const at::Tensor& grad,
137 const at::Scalar& base,
138 const at::Tensor& exponent,
139 const at::Tensor& result);
140 at::Tensor angle_backward(const at::Tensor& grad, const at::Tensor& self);
142 at::Tensor mul_tensor_backward(const Tensor& grad, T other, ScalarType self_st);
144 at::Tensor div_tensor_self_backward(
148 at::Tensor div_tensor_other_backward(
153 at::Tensor div_tensor_self_backward(
158 at::Tensor div_tensor_other_backward(
163 at::Tensor mvlgamma_backward(
164 const at::Tensor& grad,
165 const at::Tensor& self,
167 at::Tensor permute_backwards(const at::Tensor& grad, at::IntArrayRef fwd_dims);
168 at::Tensor rad2deg_backward(const at::Tensor& grad);
169 at::Tensor deg2rad_backward(const at::Tensor& grad);
170 at::Tensor unsqueeze_multiple(
171 const at::Tensor& t,
172 at::OptionalIntArrayRef opt_dim,
174 at::Tensor sum_backward(
175 const at::Tensor& grad,
176 at::SymIntArrayRef sizes,
177 at::OptionalIntArrayRef opt_dims,
179 at::Tensor sum_backward(
180 const at::Tensor& grad,
184 at::Tensor nansum_backward(
185 const at::Tensor& grad,
186 const at::Tensor& self,
187 at::OptionalIntArrayRef dims,
189 std::vector<int64_t> reverse_list(const at::IntArrayRef list);
191 at::Tensor reverse_dim(const at::Tensor& t, int64_t dim);
192 at::Tensor prod_safe_zeros_backward(
193 const at::Tensor& grad,
194 const at::Tensor& inp,
196 at::Tensor prod_backward(
197 const at::Tensor& grad,
198 const at::Tensor& input,
199 const at::Tensor& result);
200 at::Tensor prod_backward(
201 at::Tensor grad,
202 const at::Tensor& input,
203 at::Tensor result,
206 at::Tensor solve_jvp(
211 at::Tensor solve_backward_self(
212 const at::Tensor& grad,
213 const at::Tensor& self,
214 const at::Tensor& A);
215 at::Tensor solve_backward_A(
216 const at::Tensor& grad,
217 const at::Tensor& self,
218 const at::Tensor& A,
219 const at::Tensor& solution);
220 at::Tensor cumsum_backward(const at::Tensor& grad, int64_t dim);
221 at::Tensor logsumexp_backward(
222 at::Tensor grad,
223 const at::Tensor& self,
224 at::Tensor result,
225 at::IntArrayRef dim,
227 at::Tensor logsumexp_jvp(
228 const at::Tensor& self_p,
229 const at::Tensor& self_t,
232 at::Tensor safe_logsumexp_jvp(
233 const at::Tensor& self_p,
234 const at::Tensor& self_t,
237 at::Tensor logcumsumexp_backward(
238 at::Tensor grad,
239 const at::Tensor& self,
240 const at::Tensor& result,
242 at::Tensor logcumsumexp_jvp(
243 const at::Tensor& self_p,
244 const at::Tensor& self_t,
246 at::Tensor unbind_backward(const variable_list& grads, int64_t dim);
247 at::Tensor unbind_backward_nested(
251 const at::TensorOptions& options);
252 at::Tensor unbind_backward_nested_jagged(
256 at::Tensor unsqueeze_to(const at::Tensor& self, c10::SymIntArrayRef sym_sizes);
257 at::Tensor unsqueeze_to(
258 const at::Tensor& self,
261 at::Tensor unsqueeze_to(
262 const at::Tensor& self,
265 std::vector<at::Tensor> cat_tensors_backward(
266 const at::Tensor& grad,
270 std::vector<at::Tensor> stack_tensors_backward(
271 const at::Tensor& grad,
274 std::vector<at::Tensor> block_diag_backward(
275 const at::Tensor& grad,
278 at::Tensor clamp_backward(
279 const at::Tensor& grad,
280 const at::Tensor& self,
281 const std::optional<at::Scalar>& min,
282 const std::optional<at::Scalar>& max);
283 at::Tensor clamp_backward(
284 const at::Tensor& grad,
285 const at::Tensor& self,
286 const at::Tensor& min,
287 const at::Tensor& max);
288 std::tuple<at::Tensor, at::Tensor> clamp_backward_min_max(
289 const at::Tensor& grad,
290 const at::Tensor& self,
291 const at::Tensor& min,
292 const at::Tensor& max,
294 at::Tensor clamp_jvp(
301 at::SymIntArrayRef strides_or_error(
304 at::Tensor mm_mat1_backward(
307 at::SymIntArrayRef mat1_sizes,
308 at::SymIntArrayRef mat1_strides,
311 at::Tensor mm_mat2_backward(
312 const at::Tensor& grad,
313 const at::Tensor& mat1,
314 at::SymIntArrayRef sizes,
315 at::SymIntArrayRef strides,
317 const at::Scalar& alpha);
318 at::Tensor mm_mat1_sparse_backward(
319 const at::Tensor& grad,
320 const at::Tensor& mat1,
321 const at::Tensor& mat2,
322 const at::Scalar& alpha);
331 at::Tensor sparse_mask_backward(
332 const at::Tensor& grad,
333 const at::Tensor& mask,
335 at::Tensor sparse_sparse_matmul_backward(
336 const at::Tensor& grad,
337 const at::Tensor& mat1,
338 const at::Tensor& mat2,
340 at::Tensor renorm_backward(
341 const at::Tensor& grad,
342 const at::Tensor& self,
343 const at::Scalar& p,
345 const at::Scalar& maxnorm);
346 at::Tensor renorm_jvp(
347 const at::Tensor& self_p,
348 const at::Tensor& self_t,
349 const at::Scalar& p,
351 const at::Scalar& maxnorm);
352 at::Tensor repeat_backward(
353 at::Tensor grad,
354 at::SymIntArrayRef repeats,
355 at::SymIntArrayRef input_shape);
356 at::Tensor _fused_dropout_backward(
357 const at::Tensor& grad,
358 const at::Tensor& mask,
360 at::Tensor infinitely_differentiable_native_dropout_backward(
361 const at::Tensor& grad,
362 const at::Tensor& mask,
364 at::Tensor native_dropout_double_backward(
365 const at::Tensor& ggI,
366 const at::Tensor& grad,
367 const at::Tensor& mask,
369 at::Tensor evenly_distribute_backward(
370 const at::Tensor& grad,
371 const at::Tensor& input,
372 const at::Tensor& value);
375 at::Tensor var_backward(
376 at::Tensor grad,
377 const at::Tensor& self,
378 at::OptionalIntArrayRef dim,
381 at::Tensor var_jvp(
382 const at::Tensor& self_t,
383 const at::Tensor& self_p,
384 const at::Tensor& result,
385 at::OptionalIntArrayRef dim_opt,
388 at::Tensor std_backward(
389 const at::Tensor& result,
390 const at::Tensor& grad,
391 const at::Tensor& self,
392 at::OptionalIntArrayRef dim,
398 at::OptionalIntArrayRef opt_dim,
405 at::OptionalIntArrayRef dim_opt,
413 at::OptionalIntArrayRef dim_opt,
416 at::Tensor cholesky_backward(
417 const at::Tensor& grad,
419 const at::Tensor& L);
420 at::Tensor cholesky_jvp(
421 const at::Tensor& input_tangent,
422 const at::Tensor& L,
424 at::Tensor cholesky_inverse_backward(
425 const at::Tensor& grad,
426 const at::Tensor& L,
428 const at::Tensor& inverse);
429 at::Tensor cholesky_inverse_jvp(
430 const at::Tensor& F,
431 const at::Tensor& dF,
432 const at::Tensor& X,
441 at::Tensor split_with_sizes_backward(
446 const at::TensorOptions& options);
447 at::Tensor _nested_split_with_sizes_backward(
452 const at::TensorOptions& options);
453 at::Tensor split_backward(
458 const at::TensorOptions& options);
459 at::Tensor max_pool_double_backward(
460 const at::Tensor& grad,
461 const at::Tensor& indices,
463 at::Tensor error_for_max_pool2d_double_backward();
464 at::Tensor glu_double_backward(
465 const at::Tensor& grad,
466 const at::Tensor& grad_output,
467 const at::Tensor& input,
469 at::Tensor glu_double_backward_grad_output(
470 const at::Tensor& grad,
471 const at::Tensor& input,
473 at::Tensor infinitely_differentiable_silu_backward(
474 const at::Tensor& grad_output,
475 const at::Tensor& input);
476 at::Tensor infinitely_differentiable_mish_backward(
477 const at::Tensor& grad_output,
478 const at::Tensor& input);
503 at::Tensor binary_cross_entropy_with_logits_target_backward(
504 const at::Tensor& grad_output,
505 const at::Tensor& self,
506 const at::Tensor& target,
507 const std::optional<at::Tensor>& weight,
508 const std::optional<at::Tensor>& pos_weight,
510 at::Tensor log_sigmoid_double_backward(
511 const at::Tensor& grad,
512 const at::Tensor& input);
513 at::Tensor softmax_double_backward(
514 const at::Tensor& grad,
515 const at::Tensor& grad_output,
517 const at::Tensor& output);
518 at::Tensor binary_cross_entropy_double_backward(
519 const at::Tensor& grad_output,
520 const at::Tensor& grad,
521 const at::Tensor& input,
522 const at::Tensor& target,
523 const std::optional<at::Tensor>& weight,
525 at::Tensor binary_cross_entropy_double_backward_grad_output(
526 const at::Tensor& grad,
527 const at::Tensor& input,
528 const at::Tensor& target,
529 const std::optional<at::Tensor>& weight,
531 at::Tensor smooth_l1_loss_double_backward(
532 const at::Tensor& grad,
533 const at::Tensor& input,
534 const at::Tensor& target,
537 at::Tensor huber_loss_double_backward(
538 const at::Tensor& grad,
539 const at::Tensor& input,
540 const at::Tensor& target,
543 at::Tensor huber_loss_double_backward_grad_output(
544 const at::Tensor& grad,
545 const at::Tensor& grad_output,
546 const at::Tensor& input,
547 const at::Tensor& target,
550 at::Tensor mse_loss_double_backward(
551 const at::Tensor& grad,
552 const at::Tensor& input,
554 at::Tensor soft_margin_loss_double_backward(
555 const at::Tensor& grad,
556 const at::Tensor& input,
557 const at::Tensor& target,
559 at::Tensor soft_margin_loss_double_backward_grad_output(
560 const at::Tensor& grad,
561 const at::Tensor& grad_output,
562 const at::Tensor& input,
563 const at::Tensor& target,
565 at::Tensor softplus_double_backward(
566 const at::Tensor& grad,
567 const at::Tensor& input,
568 const at::Scalar& beta,
569 const at::Scalar& threshold);
570 std::tuple<at::Tensor, at::Tensor> slogdet_jvp(
571 const at::Tensor& LU,
572 const at::Tensor& pivots,
573 const at::Tensor& dA,
574 const at::Tensor& sign,
576 at::Tensor slogdet_backward(
577 const at::Tensor& grad_sign,
578 const at::Tensor& grad_logabsdet,
579 const at::Tensor& A,
580 const at::Tensor& signdet,
581 const at::Tensor& LU,
582 const at::Tensor& pivots);
583 at::Tensor log1p_backward(const at::Tensor& grad, const at::Tensor& self);
584 at::Tensor sinc_backward(const at::Tensor& grad, const at::Tensor& self);
585 at::Tensor sparse_constructor_values_backward(
586 const at::Tensor& sparse_grad_out,
587 const at::Tensor& indices);
588 at::Tensor embedding_dense_double_backward_symint(
589 const at::Tensor& grad,
590 const at::Tensor& indices,
592 at::Tensor index_backward(
593 at::Tensor zeros_like_self,
595 const at::Tensor& grad);
596 at::Tensor _cudnn_ctc_loss_backward(
597 const at::Tensor& grad_out,
598 const at::Tensor& loss,
599 const at::Tensor& raw_grad,
601 at::Tensor elu_double_backward(
625 const at::Tensor& grad,
739 at::IntArrayRef dim,
982 at::SymIntArrayRef stride,
983 at::SymIntArrayRef padding,
984 at::SymIntArrayRef dilation,
986 at::SymIntArrayRef output_padding,
996 at::SymIntArrayRef stride,
997 at::SymIntArrayRef padding,
998 at::SymIntArrayRef dilation,
1000 at::SymIntArrayRef output_padding,
1011 Tensor cat_jvp(const at::ITensorListRef& tensors, int64_t dim);
1012 Tensor block_diag_jvp(at::TensorList tensors);
1013 Tensor stack_jvp(at::TensorList tensors, int64_t dim);
1031 const at::Tensor& self,
1032 const at::Tensor& grad_output,
1033 const at::Tensor& weight,
1034 at::SymIntArrayRef padding,
1035 at::SymIntArrayRef output_padding,
1036 at::SymIntArrayRef stride,
1037 at::SymIntArrayRef dilation,
1109 at::IntArrayRef batch_sizes,
1111 const at::Tensor& workspace);