Home
last modified time | relevance | path

Searched defs:gradOutput_ (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAdaptiveAveragePooling3d.cpp232 const Tensor& gradOutput_, in adaptive_avg_pool3d_backward_out_cpu_template()
332 Tensor& adaptive_avg_pool3d_backward_out_cpu(const Tensor& gradOutput_, in adaptive_avg_pool3d_backward_out_cpu()
340 Tensor adaptive_avg_pool3d_backward_cpu(const Tensor& gradOutput_, in adaptive_avg_pool3d_backward_cpu()
H A DFractionalMaxPool3d.cpp328 const Tensor& gradOutput_, in fractional_max_pool3d_backward_out_cpu_template()
395 Tensor& fractional_max_pool3d_backward_out_cpu(const at::Tensor& gradOutput_, in fractional_max_pool3d_backward_out_cpu()
412 const at::Tensor& gradOutput_, in fractional_max_pool3d_backward_cpu()
H A DDilatedMaxPool3d.cpp250 Tensor& max_pool3d_with_indices_backward_out_cpu(const Tensor& gradOutput_, in max_pool3d_with_indices_backward_out_cpu()
274 const Tensor& gradOutput_, in max_pool3d_with_indices_backward_cpu()
H A DAdaptivePooling.h39 inline void adaptive_pool_empty_output_check(const Tensor& gradOutput_, const char* arg_name) { in adaptive_pool_empty_output_check()
H A DAdaptiveMaxPooling3d.cpp369 auto gradOutput_ = gradOutput.contiguous(); in TORCH_IMPL_FUNC() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveAveragePooling3d.cu424 const Tensor& gradOutput_, in adaptive_avg_pool3d_backward_out_cuda_template()
524 Tensor& adaptive_avg_pool3d_backward_out_cuda(const Tensor& gradOutput_, in adaptive_avg_pool3d_backward_out_cuda()
535 const Tensor& gradOutput_, in adaptive_avg_pool3d_backward_cuda()
H A DFractionalMaxPool3d.cu202 auto gradOutput_ = gradOutput; in fractional_max_pool3d_backward_out_cuda_template() local
309 Tensor& fractional_max_pool3d_backward_out_cuda(const at::Tensor& gradOutput_, in fractional_max_pool3d_backward_out_cuda()
H A DReplicationPadding.cu275 auto gradOutput_ = gradOutput; in replication_pad2d_backward_out_cuda_template() local
389 auto gradOutput_ = gradOutput; in replication_pad3d_backward_out_cuda_template() local
506 auto gradOutput_ = gradOutput; in TORCH_IMPL_FUNC() local
H A DFractionalMaxPool2d.cu238 auto gradOutput_ = gradOutput; in TORCH_IMPL_FUNC() local
H A DAdaptiveMaxPooling2d.cu347 const at::Tensor gradOutput_ = gradOutput.contiguous(); in TORCH_IMPL_FUNC() local
H A DAdaptiveMaxPooling3d.cu402 const Tensor gradOutput_ = gradOutput.contiguous(); in TORCH_IMPL_FUNC() local
H A DAdaptiveAveragePooling.cu603 const Tensor& gradOutput_, in adaptive_avg_pool2d_backward_out_cuda_template()
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py2550 gradOutput_, argument