/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
|