| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleMoreKernel.cpp | 103 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 D | AdaptiveAvgPoolKernel.cpp | 264 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 D | AdaptiveMaxPoolKernel.cpp | 352 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 D | AvgPoolKernel.cpp | 361 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 D | batch_norm_kernel.cpp | 422 …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 D | PaddingKernel.cpp | 321 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 D | MaxPoolKernel.cpp | 483 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 D | MaxUnpoolKernel.cpp | 170 auto grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in cpu_max_unpool_backward() local
|
| H A D | SoftMaxKernel.cpp | 227 scalar_t* grad_input_data = grad_input_data_base + i * dim_size; in _vec_host_softmax_backward_lastdim() local
|
| H A D | UpSampleKernel.cpp | 1935 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 D | LossMultiMargin.cpp | 153 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 D | LossMultiLabelMargin.cpp | 207 scalar_t* grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in multilabel_margin_loss_backward_out_frame() local
|
| H A D | SegmentReduce.cpp | 216 auto* grad_input_data = grad_input.mutable_data_ptr<scalar_t>(); in _segment_reduce_cpu_lengths_backward_kernel1() local
|
| H A D | LossNLL2d.cpp | 339 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 D | SegmentReduce.cu | 159 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 D | DilatedMaxPool3d.cu | 534 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 D | op_convolution_backward.cpp | 130 CTYPE* grad_input_data = nullptr; in conv2d_backward_impl() local
|