| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Pooling.cpp | 113 const Tensor& grad_output, in mkldnn_max_pool2d_backward()
125 const Tensor& grad_output, in mkldnn_max_pool3d_backward()
136 Tensor& mkldnn_avg_pool2d_backward_out(const Tensor & grad_output, in mkldnn_avg_pool2d_backward_out()
149 const Tensor& grad_output, in mkldnn_avg_pool2d_backward()
160 Tensor& mkldnn_avg_pool3d_backward_out(const Tensor & grad_output, in mkldnn_avg_pool3d_backward_out()
173 const Tensor& grad_output, in mkldnn_avg_pool3d_backward()
185 const Tensor& grad_output, in mkldnn_adaptive_avg_pool2d_backward()
291 const Tensor& grad_output, in _mkldnn_pooling_backward()
533 const Tensor& grad_output, in mkldnn_max_pool2d_backward()
554 const Tensor& grad_output, in mkldnn_max_pool3d_backward()
[all …]
|
| H A D | Linear.cpp | 33 IntArrayRef input_size, const Tensor& grad_output, const Tensor& weight) { in mkldnn_linear_backward_input()
38 const Tensor& grad_output, const Tensor& input, const Tensor& weight, bool bias_defined) { in mkldnn_linear_backward_weights()
112 IntArrayRef input_size, const Tensor& grad_output, const Tensor& weight_t){ in mkldnn_linear_backward_input()
142 const Tensor& grad_output, const Tensor& input, const Tensor& weight, bool bias_defined) { in mkldnn_linear_backward_weights()
171 const Tensor& input, const Tensor& grad_output, in mkldnn_linear_backward()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | Conv_miopen.cpp | 41 IntArrayRef input_size, const at::Tensor& grad_output, const at::Tensor& weight, in miopen_convolution_backward_input()
48 IntArrayRef weight_size, const at::Tensor& grad_output, const at::Tensor& input, in miopen_convolution_backward_weight()
55 const at::Tensor& grad_output) { in miopen_convolution_backward_bias()
60 const at::Tensor& input, const at::Tensor& grad_output, const at::Tensor& weight, in miopen_convolution_backward()
74 const at::Tensor& grad_output, const at::Tensor& weight, in miopen_convolution_transpose_backward_input()
81 IntArrayRef weight_size, const at::Tensor& grad_output, const at::Tensor& input, in miopen_convolution_transpose_backward_weight()
88 const at::Tensor& input, const at::Tensor& grad_output, const at::Tensor& weight, in miopen_convolution_transpose_backward()
102 IntArrayRef input_size, const at::Tensor& grad_output, const at::Tensor& weight, in miopen_depthwise_convolution_backward_input()
109 IntArrayRef weight_size, const at::Tensor& grad_output, const at::Tensor& input, in miopen_depthwise_convolution_backward_weight()
116 const at::Tensor& input, const at::Tensor& grad_output, const at::Tensor& weight, in miopen_depthwise_convolution_backward()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | autograd_function_db.py | 42 def backward(ctx, grad_output, grad_saved): argument
70 def backward(ctx, grad_output, grad_saved): argument
100 def backward(ctx, grad_output, grad_saved): argument
116 def backward(ctx, grad_output): argument
162 def backward(ctx, grad_output): argument
193 def backward(ctx, grad_output): argument
233 def backward(ctx, grad_output, _0, _1): argument
271 def backward(ctx, grad_output, _0, _1): argument
310 def backward(ctx, grad_output): argument
356 def backward(ctx, grad_output): argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_module_hooks.py | 282 def fn(_unused_module, grad_output): argument
953 def bw_hook(inc, h_module, grad_input, grad_output): argument
1027 def bw_fail1(self, grad_input, grad_output): argument
1030 def bw_fail2(self, grad_input, grad_output): argument
1046 def bw_hook(module, grad_input, grad_output): argument
1235 def bw_hook(inc, h_module, grad_input, grad_output): argument
1244 def bw_pre_hook(inc, h_module, grad_output): argument
1332 def bw_pre_hook(m, grad_output): argument
1336 def bw_hook(m, grad_in, grad_output): argument
1352 def bw_fail1(self, grad_input, grad_output): argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | grad.py | 11 grad_output, argument
61 grad_output, argument
109 grad_output, argument
159 grad_output, argument
207 grad_output, argument
257 grad_output, argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_autograd_function.py | 26 def backward(ctx, grad_output): argument
41 def backward(ctx, grad_output): argument
106 def backward(ctx, grad_output): argument
146 def backward(ctx, grad_output): argument
162 def backward(ctx, grad_output): argument
180 def backward(ctx, grad_output): argument
199 def backward(ctx, grad_output): argument
212 def backward(ctx, grad_output): argument
389 def backward(cls, ctx, grad_output): argument
677 def backward(ctx, grad_output): argument
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | ConvPlaceholders.cpp | 57 const at::Tensor& grad_output, in cudnn_convolution_backward_input()
72 const at::Tensor& grad_output, in cudnn_convolution_backward_weight()
87 const at::Tensor& grad_output, in cudnn_convolution_backward()
115 const at::Tensor& grad_output, in cudnn_convolution_transpose_backward_input()
130 const at::Tensor& grad_output, in cudnn_convolution_transpose_backward_weight()
145 const at::Tensor& grad_output, in cudnn_convolution_transpose_backward()
177 const at::Tensor& grad_output, in raw_cudnn_convolution_backward_input_out()
192 const Tensor& grad_output, in raw_cudnn_convolution_backward_weight_out()
|
| H A D | ConvShared.cpp | 319 TensorArg grad_output{grad_output_t, "grad_output", 1}, in cudnn_convolution_transpose_backward_input() local
368 const TensorArg& grad_output, in cudnn_convolution_backward_input()
409 const TensorArg& grad_output, in cudnn_convolution_transpose_forward()
452 TensorArg grad_output{grad_output_t, "grad_output", 1}, in cudnn_convolution_backward_input() local
593 Tensor grad_output = grad_output_t.to(input.suggest_memory_format()); in cudnn_convolution_backward() local
673 Tensor grad_output = grad_output_t.contiguous(input.suggest_memory_format()); in cudnn_convolution_transpose_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleMoreKernel.cpp | 99 auto grad_output = grad_output_.contiguous(); in cpu_upsample_nearest_backward() local
236 auto grad_output = grad_output_.contiguous(channels_last_memory_format); in cpu_upsample_nearest_backward_channels_last() local
339 const Tensor& grad_output, in upsample_nearest1d_backward_kernel_impl()
348 const Tensor& grad_output, in _upsample_nearest_exact1d_backward_kernel_impl()
357 const Tensor& grad_output, in upsample_nearest2d_backward_kernel_impl()
373 const Tensor& grad_output, in _upsample_nearest_exact2d_backward_kernel_impl()
389 const Tensor& grad_output, in upsample_nearest3d_backward_kernel_impl()
406 const Tensor& grad_output, in _upsample_nearest_exact3d_backward_kernel_impl()
430 auto grad_output = grad_output_.contiguous(); in cpu_upsample_linear_backward() local
607 auto grad_output = grad_output_.contiguous(channels_last_memory_format); in cpu_upsample_linear_backward_channels_last() local
[all …]
|
| H A D | PointwiseOpsKernel.cpp | 104 [=](BFloat16 input, BFloat16 target, BFloat16 grad_output) -> BFloat16 { in smooth_l1_backward_cpu_kernel()
115 …Vectorized<BFloat16> input, Vectorized<BFloat16> target, Vectorized<BFloat16> grad_output) -> Vect… in smooth_l1_backward_cpu_kernel()
151 [=](scalar_t input, scalar_t target, scalar_t grad_output) -> scalar_t { in smooth_l1_backward_cpu_kernel()
161 …Vectorized<scalar_t> input, Vectorized<scalar_t> target, Vectorized<scalar_t> grad_output) -> Vect… in smooth_l1_backward_cpu_kernel()
190 [=](scalar_t input, scalar_t target, scalar_t grad_output) -> scalar_t { in huber_backward_cpu_kernel()
201 …Vectorized<scalar_t> input, Vectorized<scalar_t> target, Vectorized<scalar_t> grad_output) -> Vect… in huber_backward_cpu_kernel()
|
| H A D | PaddingKernel.cpp | 317 auto grad_output = grad_output_.contiguous(); in cpu_padding_backward() local
406 auto grad_output = grad_output_.contiguous(memory_format); in cpu_padding_backward_channels_last() local
497 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in reflection_pad1d_backward_kernel_impl()
536 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in reflection_pad2d_backward_kernel_impl()
581 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in reflection_pad3d_backward_kernel_impl()
613 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in replication_pad1d_backward_kernel_impl()
644 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in replication_pad2d_backward_kernel_impl()
689 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in replication_pad3d_backward_kernel_impl()
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/ |
| H A D | functional.py | 239 def backward(ctx, grad_output): argument
285 def backward(ctx, grad_output): argument
299 def backward(ctx, grad_output): argument
314 def backward(ctx, grad_output): argument
355 def backward(ctx, grad_output): argument
426 def backward(ctx, grad_output): argument
451 def backward(ctx, grad_output): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Activation.cpp | 461 Tensor& hardtanh_backward_out(const Tensor& grad_output, const Tensor& self, const Scalar& min, con… in hardtanh_backward_out()
467 Tensor hardtanh_backward(const Tensor& grad_output, const Tensor& self, const Scalar& min, const Sc… in hardtanh_backward()
504 Tensor hardswish_backward(const Tensor& grad_output, const Tensor& self) { in hardswish_backward()
552 const Tensor& grad_output, in math_silu_backward()
559 const Tensor& grad_output, in mish_backward()
568 const Tensor& grad_output, in math_mish_backward()
654 const Tensor& grad_output, in rrelu_with_noise_backward()
785 Tensor log_sigmoid_backward_cuda(const Tensor& grad_output, const Tensor& input, const Tensor& buff… in log_sigmoid_backward_cuda()
797 Tensor log_sigmoid_backward_cpu(const Tensor& grad_output, const Tensor& input, const Tensor& buffe… in log_sigmoid_backward_cpu()
809 Tensor& log_sigmoid_backward_cuda_out(const Tensor& grad_output, const Tensor& input, in log_sigmoid_backward_cuda_out()
[all …]
|
| H A D | Loss.cpp | 402 Tensor& soft_margin_loss_backward_out(const Tensor& grad_output, const Tensor& input, const Tensor&… in soft_margin_loss_backward_out()
412 Tensor soft_margin_loss_backward(const Tensor& grad_output, const Tensor& input, const Tensor& targ… in soft_margin_loss_backward()
441 Tensor& smooth_l1_loss_backward_out(const Tensor& grad_output, const Tensor& input, const Tensor& t… in smooth_l1_loss_backward_out()
456 Tensor smooth_l1_loss_backward(const Tensor& grad_output, const Tensor& input, const Tensor& target… in smooth_l1_loss_backward()
481 Tensor huber_loss_backward(const Tensor& grad_output, const Tensor& input, const Tensor& target, in… in huber_loss_backward()
486 Tensor& huber_loss_backward_out(const Tensor& grad_output, const Tensor& input, const Tensor& targe… in huber_loss_backward_out()
498 Tensor mse_loss_backward(const Tensor& grad_output, const Tensor& input, const Tensor& target, int6… in mse_loss_backward()
503 Tensor& mse_loss_backward_out(const Tensor& grad_output, in mse_loss_backward_out()
|
| H A D | ConvolutionMM2d.cpp | 98 const Tensor& grad_output, in slow_conv2d_shape_check()
288 TensorAccessor<const scalar_t, 3> grad_output, in slow_conv2d_backward_update_grad_input_frame()
401 const Tensor grad_output = grad_output_.contiguous(memory_format); in slow_conv2d_backward_out_cpu_template() local
437 TensorAccessor<const scalar_t, 3> grad_output, in slow_conv2d_backward_weight_frame()
515 auto grad_output = grad_output_.contiguous(memory_format); in slow_conv2d_backward_weight_out_cpu_template() local
656 const Tensor& grad_output, in slow_conv2d_backward_out_cpu()
697 const Tensor& grad_output, in slow_conv2d_backward_cpu()
|
| H A D | ConvolutionMM3d.cpp | 112 const Tensor& grad_output, in slow_conv3d_shape_check()
314 TensorAccessor<const scalar_t, 4> grad_output, in slow_conv3d_backward_update_grad_input_frame()
372 const Tensor& grad_output, in slow_conv3d_backward_out_cpu_template()
467 TensorAccessor<const scalar_t, 4> grad_output, in slow_conv3d_backward_weight_frame()
495 const Tensor& grad_output, in slow_conv3d_backward_parameters_out_cpu_template()
690 static std::tuple<Tensor&, Tensor&, Tensor&> slow_conv3d_backward_out_cpu(const Tensor& grad_output, in slow_conv3d_backward_out_cpu()
735 const Tensor& grad_output, in slow_conv3d_backward_cpu()
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_autograd_funs.py | 28 def backward(ctx, grad_output): argument
51 def backward(ctx, grad_output): argument
89 def backward(ctx, grad_output): argument
102 def backward(ctx, grad_output): argument
124 def backward(ctx, grad_output): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.cpp | 203 const at::Tensor& grad_output, in compute_shape_binary_cross_entropy_backward()
261 const at::Tensor& grad_output, in compute_shape_convolution_backward()
417 const at::Tensor& grad_output, in compute_shape_embedding_dense_backward()
687 const at::Tensor& grad_output, in compute_shape_native_dropout_backward()
798 const at::Tensor& grad_output, in compute_shape_smooth_l1_loss_backward()
828 const at::Tensor& grad_output, in compute_shape_log_sigmoid_backward()
851 const at::Tensor& grad_output, in compute_shape_nll_loss2d_backward()
876 const at::Tensor& grad_output, in compute_shape_grid_sampler_2d_backward()
941 const at::Tensor& grad_output, in compute_shape__adaptive_avg_pool2d_backward()
1024 const at::Tensor& grad_output, in compute_shape__adaptive_avg_pool3d_backward()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cuh | 81 const PTA& grad_output; member
358 … const GenericPackedTensorAccessor<const input_scalar_t, 3, DefaultPtrTraits, index_t> grad_output, in batch_norm_backward_kernel()
475 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_output, in batch_norm_backward_reduce_kernel()
510 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_output, in batch_norm_backward_elemt_kernel_impl()
548 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_output, in batch_norm_backward_elemt_kernel()
570 const GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_output, in batch_norm_backward_elemt_kernel()
628 auto grad_output = get_packed_accessor< in batch_norm_backward_cuda_template() local
804 auto grad_output = get_packed_accessor< in batch_norm_backward_reduce_cuda_template() local
852 auto grad_output = get_packed_accessor< in batch_norm_backward_elemt_cuda_template() local
903 auto grad_output = get_packed_accessor< in batch_norm_backward_elemt_cuda_template() local
[all …]
|
| H A D | ReflectionPad.cu | 111 scalar_t * grad_input, const scalar_t * grad_output, in reflection_pad1d_backward_out_kernel()
146 scalar_t * grad_input, const scalar_t * grad_output, in reflection_pad2d_backward_out_kernel()
244 PackedTensorAccessor64<const scalar_t, 5> grad_output, in reflection_pad3d_backward_out_kernel()
398 Tensor grad_output = grad_output_.contiguous(); in reflection_pad2d_backward_out_template() local
514 Tensor grad_output = grad_output_.contiguous(); in TORCH_IMPL_FUNC() local
542 Tensor& reflection_pad2d_backward_out_cuda(const Tensor& grad_output, in reflection_pad2d_backward_out_cuda()
557 const Tensor& grad_output, in reflection_pad2d_backward_cuda()
|
| H A D | ConvolutionMM2d.cu | 25 const Tensor& input, const Tensor& grad_output, in slow_conv2d_shape_check()
218 const Tensor &grad_output, in slow_conv2d_backward()
293 const Tensor &grad_output, in slow_conv2d_grad_weight()
430 auto grad_output = grad_output_.expect_contiguous(); in slow_conv2d_backward_out_cuda() local
467 const Tensor& grad_output, in slow_conv2d_backward_cuda()
|
| H A D | NaiveConvolutionTranspose2d.cu | 31 const Tensor& grad_output, in slow_conv_transpose2d_shape_check()
382 Tensor grad_output = grad_output_.contiguous(); in slow_conv_transpose2d_backward_out_cuda_template() local
558 Tensor grad_output = grad_output_.contiguous(); in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
707 …tuple<Tensor&, Tensor&, Tensor&> slow_conv_transpose2d_backward_out_cuda(const Tensor& grad_output, in slow_conv_transpose2d_backward_out_cuda()
760 const Tensor& grad_output, in slow_conv_transpose2d_backward_cuda()
|
| /aosp_15_r20/external/pytorch/torch/masked/maskedtensor/ |
| H A D | _ops_refs.py | 62 def backward(ctx, grad_output): argument
82 def backward(ctx, grad_output): argument
112 def backward(ctx, grad_output): argument
138 def backward(ctx, grad_output): argument
150 def backward(ctx, grad_output): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/ |
| H A D | NestedTensorBackward.cpp | 41 const Tensor& grad_output, in nested_linear_backward()
98 Tensor grad_output = in nested_softmax_backward() local
185 Tensor threshold_backwards_nested(const Tensor& grad_output, const Tensor& input, const Scalar& thr… in threshold_backwards_nested()
191 Tensor silu_backward_nested(const Tensor& grad_output, const Tensor& self){ in silu_backward_nested()
|