Home
last modified time | relevance | path

Searched defs:grad_input (Results 1 – 25 of 102) sorted by relevance

12345

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLoss.cpp315 Tensor grad_input = at::empty_like(input); in binary_cross_entropy_backward_cpu() local
320 …t Tensor& target, const std::optional<Tensor>& weight_opt, int64_t reduction, Tensor& grad_input) { in binary_cross_entropy_backward_out_cpu()
402 …r& grad_output, const Tensor& input, const Tensor& target, int64_t reduction, Tensor& grad_input) { in soft_margin_loss_backward_out()
413 auto grad_input = at::empty({0}, input.options()); in soft_margin_loss_backward() local
441 …t, const Tensor& input, const Tensor& target, int64_t reduction, double beta, Tensor& grad_input) { in smooth_l1_loss_backward_out()
457 auto grad_input = at::zeros_like(input, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in smooth_l1_loss_backward() local
482 auto grad_input = at::zeros_like(input, MemoryFormat::Contiguous); in huber_loss_backward() local
486 …, const Tensor& input, const Tensor& target, int64_t reduction, double delta, Tensor& grad_input) { in huber_loss_backward_out()
499 Tensor grad_input = at::zeros_like(input, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in mse_loss_backward() local
504 const Tensor& input, const Tensor& target, int64_t reduction, Tensor& grad_input) { in mse_loss_backward_out()
H A DActivation.cpp461 …ensor& grad_output, const Tensor& self, const Scalar& min, const Scalar& max, Tensor& grad_input) { in hardtanh_backward_out()
505 Tensor grad_input; in hardswish_backward() local
561 Tensor grad_input = at::empty({0}, input.options()); in mish_backward() local
786 auto grad_input = at::empty_like(grad_output); in log_sigmoid_backward_cuda() local
798 auto grad_input = at::empty_like(grad_output); in log_sigmoid_backward_cpu() local
810 const Tensor& buffer, Tensor& grad_input) { in log_sigmoid_backward_cuda_out()
823 Tensor& grad_input) { in log_sigmoid_backward_cpu_out()
H A DLossMultiLabelMargin.cpp158 Tensor& grad_input, in multilabel_margin_loss_backward_out_frame()
227 Tensor& grad_input, in multilabel_margin_loss_backward_out_cpu_template()
302 Tensor& grad_input) { in multilabel_margin_loss_backward_cpu_out()
314 auto grad_input = at::zeros_like(self, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in multilabel_margin_loss_backward_cpu() local
H A DLossNLL2d.cpp284 Tensor& grad_input, in nll_loss2d_backward_out_frame()
367 Tensor& grad_input, in nll_loss2d_backward_out_cpu_template()
444 Tensor& grad_input) { in nll_loss2d_backward_out_cpu()
472 auto grad_input = at::zeros_like(self); in nll_loss2d_backward_cpu() local
H A DReflectionPad.cpp270 Tensor &grad_input, const Tensor &grad_output, in reflection_pad2d_backward_out_template()
353 Tensor& grad_input) { in reflection_pad2d_backward_out_cpu()
365 auto grad_input = at::zeros_like(input, input.suggest_memory_format()); in reflection_pad2d_backward_cpu() local
H A DLossMultiMargin.cpp211 Tensor& grad_input, in multi_margin_loss_backward_out_cpu_template()
300 auto grad_input = at::empty({0}, input.options()); in multi_margin_loss_cpu_backward() local
319 Tensor& grad_input) { in multi_margin_loss_cpu_backward_out()
H A DAdaptiveAveragePooling.cpp62 Tensor& grad_input, in adaptive_avg_pool2d_backward_out_cpu_template()
146 auto grad_input = at::empty({0}, input.options()); in adaptive_avg_pool2d_backward_cpu() local
H A DNaiveDilatedConvolution.cpp178 Tensor& grad_input, in slow_conv_dilated_all_cpu_template()
663 Tensor grad_input = in slow_conv_dilated2d_backward_cpu() local
719 Tensor grad_input = in slow_conv_dilated3d_backward_cpu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp100 auto grad_input = grad_input_.contiguous(); in cpu_upsample_nearest_backward() local
237 auto grad_input = grad_input_.contiguous(channels_last_memory_format); in cpu_upsample_nearest_backward_channels_last() local
338 const Tensor& grad_input, in upsample_nearest1d_backward_kernel_impl()
347 const Tensor& grad_input, in _upsample_nearest_exact1d_backward_kernel_impl()
356 const Tensor& grad_input, in upsample_nearest2d_backward_kernel_impl()
372 const Tensor& grad_input, in _upsample_nearest_exact2d_backward_kernel_impl()
388 const Tensor& grad_input, in upsample_nearest3d_backward_kernel_impl()
405 const Tensor& grad_input, in _upsample_nearest_exact3d_backward_kernel_impl()
431 auto grad_input = grad_input_.contiguous(); in cpu_upsample_linear_backward() local
608 auto grad_input = grad_input_.contiguous(channels_last_memory_format); in cpu_upsample_linear_backward_channels_last() local
[all …]
H A DPaddingKernel.cpp318 auto grad_input = grad_input_.contiguous(); in cpu_padding_backward() local
405 auto grad_input = grad_input_.contiguous(memory_format); in cpu_padding_backward_channels_last() local
497 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in reflection_pad1d_backward_kernel_impl()
536 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in reflection_pad2d_backward_kernel_impl()
581 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in reflection_pad3d_backward_kernel_impl()
613 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in replication_pad1d_backward_kernel_impl()
644 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in replication_pad2d_backward_kernel_impl()
689 const Tensor& grad_input, const Tensor& grad_output, IntArrayRef padding) { in replication_pad3d_backward_kernel_impl()
H A DAdaptiveAvgPoolKernel.cpp261 auto grad_input = grad_input_.contiguous(); in cpu_adaptive_avg_pool2d_backward() local
311 auto grad_input = grad_input_.contiguous(memory_format); in cpu_adaptive_avg_pool2d_backward_channels_last() local
391 Tensor& grad_input, in adapative_avg_pool2d_backward_kernel_impl()
686 auto grad_input = grad_input_.contiguous(); in cpu_adaptive_avg_pool3d_backward() local
745 auto grad_input = grad_input_.contiguous(memory_format); in cpu_adaptive_avg_pool3d_backward_channels_last() local
835 Tensor& grad_input, in adapative_avg_pool3d_backward_kernel_impl()
H A DAdaptiveMaxPoolKernel.cpp348 auto grad_input = grad_input_.contiguous(); in cpu_adaptive_max_pool2d_backward() local
395 auto grad_input = grad_input_.contiguous(memory_format); in cpu_adaptive_max_pool2d_backward_channels_last() local
461 const Tensor& grad_input, in adaptive_max_pool2d_backward_kernel_impl()
838 auto grad_input = grad_input_.contiguous(); in cpu_adaptive_max_pool3d_backward() local
889 auto grad_input = grad_input_.contiguous(memory_format); in cpu_adaptive_max_pool3d_backward_channels_last() local
959 const Tensor& grad_input, in adaptive_max_pool3d_backward_kernel_impl()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DLoss.cu34 void binary_cross_entropy_backward_out_kernel(Tensor& grad_input, const Tensor& grad, const Tensor&… in binary_cross_entropy_backward_out_kernel()
129 Tensor grad_input = at::empty_like(input); in binary_cross_entropy_backward_cuda() local
134 …t Tensor& target, const std::optional<Tensor>& weight_opt, int64_t reduction, Tensor& grad_input) { in binary_cross_entropy_backward_out_cuda()
412 PackedTensorAccessor64<scalar_t, 2> grad_input, in nll_loss_backward_no_reduce_cuda_kernel()
430 scalar_t *grad_input, in nll_loss_backward_reduce_cuda_kernel_1d()
453 scalar_t* grad_input, in nll_loss_backward_reduce_cuda_kernel_2d()
492 auto grad_input = *grad_input_.expect_contiguous(); in nll_loss_backward_out_cuda_template() local
H A DReflectionPad.cu111 scalar_t * grad_input, const scalar_t * grad_output, in reflection_pad1d_backward_out_kernel()
146 scalar_t * grad_input, const scalar_t * grad_output, in reflection_pad2d_backward_out_kernel()
243 PackedTensorAccessor64<scalar_t, 5> grad_input, in reflection_pad3d_backward_out_kernel()
357 Tensor &grad_input, const Tensor &grad_output_, in reflection_pad2d_backward_out_template()
545 Tensor& grad_input) { in reflection_pad2d_backward_out_cuda()
563 auto grad_input = at::zeros_like(input, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in reflection_pad2d_backward_cuda() local
H A DNormalization.cuh359 GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_input, in batch_norm_backward_kernel()
516 GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_input, in batch_norm_backward_elemt_kernel_impl()
554 GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_input, in batch_norm_backward_elemt_kernel()
576 GenericPackedTensorAccessor<input_scalar_t, 3, DefaultPtrTraits, index_t> grad_input, in batch_norm_backward_elemt_kernel()
630 auto grad_input = packed_accessor_or_dummy< in batch_norm_backward_cuda_template() local
850 auto grad_input = get_packed_accessor< in batch_norm_backward_elemt_cuda_template() local
901 auto grad_input = get_packed_accessor< in batch_norm_backward_elemt_cuda_template() local
1332 scalar_t* __restrict__ grad_input, in batch_norm_backward_elemt_channels_last_kernel_impl()
1387 scalar_t* __restrict__ grad_input, in batch_norm_backward_elemt_channels_last_kernel()
1416 scalar_t* __restrict__ grad_input, in batch_norm_backward_elemt_channels_last_kernel()
[all …]
H A DMaxUnpooling.cu404 Tensor& grad_input) { in max_unpooling2d_backward_out_cuda()
494 auto grad_input = at::empty_like(self, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in max_unpooling2d_backward_cuda() local
506 Tensor& grad_input) { in max_unpooling3d_backward_out_cuda()
606 auto grad_input = at::empty_like(self, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in max_unpooling3d_backward_cuda() local
H A DMultiLabelMarginCriterion.cu310 Tensor& grad_input) { in multilabel_margin_loss_backward_cuda_out_template()
422 Tensor& grad_input) { in multilabel_margin_loss_backward_cuda_out()
434 auto grad_input = at::zeros_like(self, LEGACY_CONTIGUOUS_MEMORY_FORMAT); in multilabel_margin_loss_backward_cuda() local
H A DActivation.cpp30 int64_t dim, Tensor& grad_input) { in glu_backward_cuda_out()
71 auto grad_input = at::empty({0}, input.options()); in glu_backward_cuda() local
H A DNLLLoss2d.cu355 Tensor& grad_input, in nll_loss2d_backward_out_cuda_template()
501 Tensor& grad_input) { in nll_loss2d_backward_out_cuda()
522 auto grad_input = at::empty_like(self); in nll_loss2d_backward_cuda() local
H A DNaiveDilatedConvolution.cu186 Tensor& grad_input, in slow_conv_dilated_all_cuda_template()
480 Tensor grad_input = in slow_conv_dilated2d_backward_cuda() local
586 Tensor grad_input = in slow_conv_dilated3d_backward_cuda() local
H A DGridSampler.cpp49 Tensor grad_input = ([&]() { in grid_sampler_2d_backward_cuda() local
68 Tensor grad_input = ([&]() { in grid_sampler_3d_backward_cuda() local
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_module_hooks.py953 def bw_hook(inc, h_module, grad_input, grad_output): argument
1027 def bw_fail1(self, grad_input, grad_output): argument
1030 def bw_fail2(self, grad_input, grad_output): argument
1046 def bw_hook(module, grad_input, grad_output): argument
1235 def bw_hook(inc, h_module, grad_input, grad_output): argument
1352 def bw_fail1(self, grad_input, grad_output): argument
1355 def bw_fail2(self, grad_input, grad_output): argument
1406 def hook(mod, grad_input, grad_output): argument
1457 def hook(mod, grad_input, grad_output): argument
1478 def hook(mod, grad_input, grad_output): argument
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesConvolution.cpp185 const auto grad_input = reshape_dim_outof(1, batch_size, std::get<0>(result)); in convolution_backward_input_batch_rule() local
196 const auto grad_input = reshape_dim_outof(0, batch_size, std::get<0>(result)); in convolution_backward_input_batch_rule() local
209 const auto grad_input = reshape_dim_outof(1, batch_size, std::get<0>(result)); in convolution_backward_input_batch_rule() local
212 Tensor grad_input; in convolution_backward_input_batch_rule() local
413 const auto grad_input = output_mask[0] ? in convolution_backward_plumbing() local
423 Tensor grad_input; in convolution_backward_plumbing() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_convolution_backward.cpp38 Tensor& grad_input, in check_convolution_backward_args()
103 Tensor& grad_input, in conv2d_backward_impl()
242 Tensor& grad_input, in convolution_backward_out()
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_convolution_backward_test.cpp40 Tensor& grad_input, in op_convolution_backward_out()
173 auto grad_input = tf.zeros({2, 6, 7, 5}); in TEST_F() local

12345