Home
last modified time | relevance | path

Searched defs:grad_output_ (Results 1 – 25 of 39) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesConvolution.cpp142 const Tensor& grad_output_, std::array<bool, 3> output_mask) { in compute_grad_bias()
179 const auto grad_output_ = reshape_dim_into(*grad_output_bdim, 1, grad_output); in convolution_backward_input_batch_rule() local
191 const auto grad_output_ = reshape_dim_into(*grad_output_bdim, 0, grad_output); in convolution_backward_input_batch_rule() local
259 const auto grad_output_ = reshape_dim_into(*grad_output_bdim, 1, grad_output); in convolution_backward_weight_batch_rule() local
273 const auto grad_output_ = reshape_dim_into(*grad_output_bdim, 1, grad_output); in convolution_backward_weight_batch_rule() local
283 auto grad_output_ = moveBatchDimToFront(grad_output, grad_output_bdim); // BN(GO) in convolution_backward_weight_batch_rule() local
362 const Tensor& grad_output_, const Tensor& input_, const Tensor& weight_, in convolution_backward_plumbing()
H A DBatchRulesModules.cpp147 auto grad_output_ = moveBatchDimToFront(grad_output, grad_output_bdim); in grid_sample_backward_helper_in() local
263 auto grad_output_ = reshape_dim_into(*grad_output_bdim, 0, grad_output); in apply() local
H A DBatchRulesReduceOps.cpp268 auto grad_output_ = moveBatchDimToFront(grad_output, grad_output_bdim); in _softmax_backward_batch_rule() local
299 auto grad_output_ = moveBatchDimToFront(grad_output, grad_output_bdim); in _log_softmax_backward_batch_rule() local
H A DBatchRulesActivation.cpp44 …const auto grad_output_ = ensure_has_bdim(moveBatchDimToFront(grad_output, grad_output_bdim), grad… in glu_backward_batch_rule() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp94 const Tensor& grad_output_, in cpu_upsample_nearest_backward()
227 const Tensor& grad_output_, in cpu_upsample_nearest_backward_channels_last()
424 const Tensor& grad_output_, in cpu_upsample_linear_backward()
597 const Tensor& grad_output_, in cpu_upsample_linear_backward_channels_last()
H A DAdaptiveAvgPoolKernel.cpp259 const Tensor& grad_output_) { in cpu_adaptive_avg_pool2d_backward()
309 const Tensor& grad_output_) { in cpu_adaptive_avg_pool2d_backward_channels_last()
684 const Tensor& grad_output_) { in cpu_adaptive_avg_pool3d_backward()
743 const Tensor& grad_output_) { in cpu_adaptive_avg_pool3d_backward_channels_last()
H A DAdaptiveMaxPoolKernel.cpp344 const Tensor& grad_output_, in cpu_adaptive_max_pool2d_backward()
390 const Tensor& grad_output_, in cpu_adaptive_max_pool2d_backward_channels_last()
834 const Tensor& grad_output_, in cpu_adaptive_max_pool3d_backward()
884 const Tensor& grad_output_, in cpu_adaptive_max_pool3d_backward_channels_last()
H A DAvgPoolKernel.cpp351 const Tensor& grad_output_, in cpu_avg_pool2d_backward()
419 const Tensor& grad_output_, in cpu_avg_pool2d_backward_channels_last()
911 const Tensor& grad_output_, in cpu_avg_pool3d_backward()
988 const Tensor& grad_output_, in cpu_avg_pool3d_backward_channels_last()
H A DPaddingKernel.cpp314 const Tensor& grad_output_, in cpu_padding_backward()
398 const Tensor& grad_output_, in cpu_padding_backward_channels_last()
H A DMaxPoolKernel.cpp475 const Tensor& grad_output_, in cpu_max_pool_backward()
540 const Tensor& grad_output_, in cpu_max_pool_backward_channels_last()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveDilatedConvolution.cpp656 const Tensor grad_output_ = in slow_conv_dilated2d_backward_cpu() local
712 const Tensor grad_output_ = in slow_conv_dilated3d_backward_cpu() local
H A DNaiveConvolutionTranspose2d.cpp391 const Tensor& grad_output_, in slow_conv_transpose2d_backward_out_cpu_template()
589 const Tensor& grad_output_, in slow_conv_transpose2d_acc_grad_parameters_cpu()
H A DNaiveConvolutionTranspose3d.cpp393 const Tensor& grad_output_, in slow_conv_transpose3d_backward_out_cpu_template()
597 const Tensor& grad_output_, in slow_conv_transpose3d_acc_grad_parameters_cpu()
H A DConvolutionMM2d.cpp360 const Tensor& grad_output_, in slow_conv2d_backward_out_cpu_template()
485 const Tensor& grad_output_, in slow_conv2d_backward_weight_out_cpu_template()
H A DUpSampleBicubic2d.cpp184 const Tensor& grad_output_, in upsample_bicubic2d_backward_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNaiveDilatedConvolution.cu473 const Tensor grad_output_ = in slow_conv_dilated2d_backward_cuda() local
579 const Tensor grad_output_ = in slow_conv_dilated3d_backward_cuda() local
H A DNaiveConvolutionTranspose2d.cu305 const Tensor& grad_output_, in slow_conv_transpose2d_backward_out_cuda_template()
483 const Tensor& grad_output_, in slow_conv_transpose2d_acc_grad_parameters_cuda_template()
H A DMaxUnpooling.cu400 at::Tensor& max_unpooling2d_backward_out_cuda(const Tensor& grad_output_, in max_unpooling2d_backward_out_cuda()
500 at::Tensor& max_unpooling3d_backward_out_cuda(const Tensor& grad_output_, in max_unpooling3d_backward_out_cuda()
H A DNaiveConvolutionTranspose3d.cu396 const Tensor& grad_output_, in slow_conv_transpose3d_backward_out_cuda_template()
606 const Tensor& grad_output_, in slow_conv_transpose3d_acc_grad_parameters_cuda()
H A DReflectionPad.cu357 Tensor &grad_input, const Tensor &grad_output_, in reflection_pad2d_backward_out_template()
643 auto grad_output_ = grad_output; in TORCH_IMPL_FUNC() local
H A DUpSampleBilinear2d.cu361 const Tensor& grad_output_, in upsample_bilinear2d_backward_out_cuda_template()
771 const Tensor& grad_output_, in upsample_gen2d_aa_backward_out_cuda_template()
H A DUpSampleNearest1d.cu149 const Tensor& grad_output_, in upsample_nearest1d_backward_out_cuda_template()
H A DUpSampleLinear1d.cu163 const Tensor& grad_output_, in upsample_linear1d_backward_out_cuda_template()
H A DUpSampleBicubic2d.cu224 const Tensor& grad_output_, in upsample_bicubic2d_backward_out_cuda_template()
H A DMultiLabelMarginCriterion.cu322 auto grad_output_ = grad_output.contiguous(); in multilabel_margin_loss_backward_cuda_out_template() local

12