Home
last modified time | relevance | path

Searched defs:grad_input_ (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesViews.cpp324 auto grad_input_ = moveBatchDimToFront(grad_input, grad_input_bdim); in diagonal_backward_batch_rule() local
396 auto grad_input_ = moveBatchDimToFront(grad_input, grad_input_bdim); in select_backward_batch_rule() local
409 auto grad_input_ = moveBatchDimToFront(grad_input, grad_input_bdim); in slice_backward_batch_rule() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp93 const Tensor& grad_input_, in cpu_upsample_nearest_backward()
226 const Tensor& grad_input_, in cpu_upsample_nearest_backward_channels_last()
423 const Tensor& grad_input_, in cpu_upsample_linear_backward()
596 const Tensor& grad_input_, in cpu_upsample_linear_backward_channels_last()
H A DAdaptiveAvgPoolKernel.cpp258 Tensor& grad_input_, in cpu_adaptive_avg_pool2d_backward()
308 Tensor& grad_input_, in cpu_adaptive_avg_pool2d_backward_channels_last()
683 Tensor& grad_input_, in cpu_adaptive_avg_pool3d_backward()
742 Tensor& grad_input_, in cpu_adaptive_avg_pool3d_backward_channels_last()
H A DAdaptiveMaxPoolKernel.cpp343 const Tensor& grad_input_, in cpu_adaptive_max_pool2d_backward()
389 const Tensor& grad_input_, in cpu_adaptive_max_pool2d_backward_channels_last()
833 const Tensor& grad_input_, in cpu_adaptive_max_pool3d_backward()
883 const Tensor& grad_input_, in cpu_adaptive_max_pool3d_backward_channels_last()
H A DAvgPoolKernel.cpp350 const Tensor& grad_input_, in cpu_avg_pool2d_backward()
418 const Tensor& grad_input_, in cpu_avg_pool2d_backward_channels_last()
910 const Tensor& grad_input_, in cpu_avg_pool3d_backward()
987 const Tensor& grad_input_, in cpu_avg_pool3d_backward_channels_last()
H A DPaddingKernel.cpp313 const Tensor& grad_input_, in cpu_padding_backward()
397 const Tensor& grad_input_, in cpu_padding_backward_channels_last()
H A DMaxPoolKernel.cpp474 const Tensor& grad_input_, in cpu_max_pool_backward()
539 const Tensor& grad_input_, in cpu_max_pool_backward_channels_last()
H A DMaxUnpoolKernel.cpp163 Tensor& grad_input_, in cpu_max_unpool_backward()
H A DUpSampleKernel.cpp1924 const Tensor& grad_input_, in cpu_upsample_genNd_backward_aa()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveDilatedConvolution.cpp669 Tensor grad_input_ = in slow_conv_dilated2d_backward_cpu() local
725 Tensor grad_input_ = in slow_conv_dilated3d_backward_cpu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNaiveDilatedConvolution.cu486 Tensor grad_input_ = in slow_conv_dilated2d_backward_cuda() local
592 Tensor grad_input_ = in slow_conv_dilated3d_backward_cuda() local
H A DMultiMarginLoss.cu300 int64_t reduction, Tensor &grad_input_) { in multi_margin_loss_cuda_backward_out()
H A DUpSampleTrilinear3d.cu299 const Tensor& grad_input_, in upsample_trilinear3d_backward_out_cuda_template()
H A DLoss.cu482 const Tensor& grad_input_, in nll_loss_backward_out_cuda_template()
H A DReflectionPad.cu642 auto grad_input_ = grad_input; in TORCH_IMPL_FUNC() local
H A DDepthwiseConv3d.cu550 Tensor grad_input_ = in _depthwise_3d_backward_cuda_out() local
H A DNormalization.cuh608 Tensor grad_input_; in batch_norm_backward_cuda_template() local