Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp103 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_upsample_nearest_backward() local
240 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_upsample_nearest_backward_channels_last() local
434 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_upsample_linear_backward() local
611 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_upsample_linear_backward_channels_last() local
H A DAdaptiveAvgPoolKernel.cpp264 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_adaptive_avg_pool2d_backward() local
314 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_adaptive_avg_pool2d_backward_channels_last() local
689 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_adaptive_avg_pool3d_backward() local
748 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_adaptive_avg_pool3d_backward_channels_last() local
H A DAdaptiveMaxPoolKernel.cpp352 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_adaptive_max_pool2d_backward() local
399 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_adaptive_max_pool2d_backward_channels_last() local
842 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_adaptive_max_pool3d_backward() local
893 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_adaptive_max_pool3d_backward_channels_last() local
H A DAvgPoolKernel.cpp361 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_avg_pool2d_backward() local
429 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_avg_pool2d_backward_channels_last() local
921 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_avg_pool3d_backward() local
998 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_avg_pool3d_backward_channels_last() local
H A Dbatch_norm_kernel.cpp422 …scalar_t* grad_input_data = grad_input.defined() ? grad_input.mutable_data_ptr<scalar_t>() : nullp… in batch_norm_cpu_backward_contiguous_impl() local
543 …scalar_t* grad_input_data = grad_input.defined() ? grad_input.mutable_data_ptr<scalar_t>() : nullp… in batch_norm_cpu_backward_channels_last_impl() local
1012 …scalar_t* grad_input_data = grad_input.defined() ? grad_input.mutable_data_ptr<scalar_t>() : nullp… in batch_norm_cpu_backward_contiguous_internal() local
1134 …scalar_t* grad_input_data = grad_input.defined() ? grad_input.mutable_data_ptr<scalar_t>() : nullp… in batch_norm_cpu_backward_channels_last_internal() local
H A DPaddingKernel.cpp321 auto grad_input_data = grad_input.data_ptr<scalar_t>(); in cpu_padding_backward() local
408 auto grad_input_data = grad_input.data_ptr<scalar_t>(); in cpu_padding_backward_channels_last() local
H A DMaxPoolKernel.cpp483 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_max_pool_backward() local
554 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_max_pool_backward_channels_last() local
H A DMaxUnpoolKernel.cpp170 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_max_unpool_backward() local
H A DSoftMaxKernel.cpp227 scalar_t* grad_input_data = grad_input_data_base + i * dim_size; in _vec_host_softmax_backward_lastdim() local
H A DUpSampleKernel.cpp1935 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_upsample_genNd_backward_aa() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLossMultiMargin.cpp153 scalar_t* grad_input_data, in multi_margin_loss_backward_cpu_kernel()
238 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in multi_margin_loss_backward_out_cpu_template() local
H A DLossMultiLabelMargin.cpp207 scalar_t* grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in multilabel_margin_loss_backward_out_frame() local
H A DSegmentReduce.cpp216 auto* grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in _segment_reduce_cpu_lengths_backward_kernel1() local
H A DLossNLL2d.cpp339 scalar_t* grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in nll_loss2d_backward_out_frame() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSegmentReduce.cu159 scalar_t* grad_input_data, in segment_reduce_backward_kernel()
325 auto* grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in _segment_reduce_lengths_offsets_backward_cuda_kernel() local
H A DDilatedMaxPool3d.cu534 scalar_t *grad_input_data = work_grad_input.mutable_data_ptr<scalar_t>(); in max_pool3d_with_indices_backward_out_cuda_template() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_convolution_backward.cpp130 CTYPE* grad_input_data = nullptr; in conv2d_backward_impl() local