Home
last modified time | relevance | path

Searched refs:adaptive_pool_empty_output_check (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAdaptivePooling.h39 inline void adaptive_pool_empty_output_check(const Tensor& gradOutput_, const char* arg_name) { in adaptive_pool_empty_output_check() function
H A DAdaptiveMaxPooling2d.cpp62 at::native::adaptive_pool_empty_output_check(grad_output, "adaptive_max_pool2d_backward"); in TORCH_META_FUNC()
H A DAdaptiveAveragePooling3d.cpp237 adaptive_pool_empty_output_check(gradOutput_, "adaptive_avg_pool3d_backward"); in adaptive_avg_pool3d_backward_out_cpu_template()
H A DAdaptiveMaxPooling3d.cpp69 at::native::adaptive_pool_empty_output_check(gradOutput, "adaptive_max_pool3d_backward"); in TORCH_META_FUNC()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveAveragePooling3d.cu430 adaptive_pool_empty_output_check(gradOutput_, "adaptive_avg_pool3d_backward"); in adaptive_avg_pool3d_backward_out_cuda_template()
H A DAdaptiveAveragePooling.cu610 adaptive_pool_empty_output_check(gradOutput_, "adaptive_avg_pool2d_backward"); in adaptive_avg_pool2d_backward_out_cuda_template()