Home
last modified time | relevance | path

Searched defs:grad_output_data (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp102 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_upsample_nearest_backward() local
239 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_upsample_nearest_backward_channels_last() local
433 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_upsample_linear_backward() local
610 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_upsample_linear_backward_channels_last() local
H A DAdaptiveAvgPoolKernel.cpp263 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_adaptive_avg_pool2d_backward() local
315 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_adaptive_avg_pool2d_backward_channels_last() local
688 auto grad_output_data = grad_output.data_ptr<scalar_t>(); in cpu_adaptive_avg_pool3d_backward() local
749 auto grad_output_data = grad_output.data_ptr<scalar_t>(); in cpu_adaptive_avg_pool3d_backward_channels_last() local
H A DAdaptiveMaxPoolKernel.cpp350 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_adaptive_max_pool2d_backward() local
400 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_adaptive_max_pool2d_backward_channels_last() local
840 auto grad_output_data = grad_output.data_ptr<scalar_t>(); in cpu_adaptive_max_pool3d_backward() local
894 auto grad_output_data = grad_output.data_ptr<scalar_t>(); in cpu_adaptive_max_pool3d_backward_channels_last() local
H A DAvgPoolKernel.cpp360 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_avg_pool2d_backward() local
430 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_avg_pool2d_backward_channels_last() local
920 auto grad_output_data = grad_output.data_ptr<scalar_t>(); in cpu_avg_pool3d_backward() local
999 auto grad_output_data = grad_output.data_ptr<scalar_t>(); in cpu_avg_pool3d_backward_channels_last() local
H A Dbatch_norm_kernel.cpp419 const scalar_t* grad_output_data = grad_output.const_data_ptr<scalar_t>(); in batch_norm_cpu_backward_contiguous_impl() local
540 const scalar_t* grad_output_data = grad_output.const_data_ptr<scalar_t>(); in batch_norm_cpu_backward_channels_last_impl() local
1009 const scalar_t* grad_output_data = grad_output.const_data_ptr<scalar_t>(); in batch_norm_cpu_backward_contiguous_internal() local
1131 const scalar_t* grad_output_data = grad_output.const_data_ptr<scalar_t>(); in batch_norm_cpu_backward_channels_last_internal() local
H A DPaddingKernel.cpp320 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_padding_backward() local
409 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_padding_backward_channels_last() local
H A DMaxPoolKernel.cpp481 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_max_pool_backward() local
555 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_max_pool_backward_channels_last() local
H A DMaxUnpoolKernel.cpp168 auto grad_output_data = grad_output.data_ptr<scalar_t>(); in cpu_max_unpool_backward() local
H A DUpSampleKernel.cpp1934 auto grad_output_data = grad_output.const_data_ptr<scalar_t>(); in cpu_upsample_genNd_backward_aa() local
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_convolution_backward_test.cpp70 std::vector<float> grad_output_data = { in TEST_F() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_convolution_backward.cpp126 const CTYPE* grad_output_data = grad_output.const_data_ptr<CTYPE>(); in conv2d_backward_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc887 std::unique_ptr<GlobalData> grad_output_data = in XLA_TEST_P() local