Searched defs:divide_factor (Results 1 – 7 of 7) sorted by relevance
74 int64_t divide_factor = 0; in cpu_avg_pool2d() local150 int64_t divide_factor = 0; in cpu_avg_pool2d_channels_last() local269 int64_t divide_factor = 0; in cpu_avg_pool2d_channels_last() local389 int64_t divide_factor = 0; in cpu_avg_pool2d_backward() local458 int64_t divide_factor = 0; in cpu_avg_pool2d_backward_channels_last() local614 int64_t divide_factor = 0; in cpu_avg_pool3d() local699 int64_t divide_factor = 0; in cpu_avg_pool3d_channels_last() local827 int64_t divide_factor = 0; in cpu_avg_pool3d_channels_last() local955 int64_t divide_factor = 0; in cpu_avg_pool3d_backward() local1034 int64_t divide_factor = 0; in cpu_avg_pool3d_backward_channels_last() local
73 accscalar_t divide_factor; in avg_pool3d_cuda_update_output() local142 accscalar_t divide_factor; in avg_pool3d_cuda_update_output() local247 accscalar_t divide_factor; in avg_pool3d_cuda_update_grad_input_atomic() local305 accscalar_t divide_factor; in avg_pool3d_cuda_update_grad_input() local532 accscalar_t divide_factor; in TORCH_IMPL_FUNC() local
67 int divide_factor; in avg_pool2d_out_cuda_frame() local116 int divide_factor; in avg_pool2d_out_cuda_frame_nhwc() local169 int divide_factor; in avg_pool2d_backward_out_cuda_frame() local224 int divide_factor; in avg_pool2d_backward_out_cuda_frame_nhwc() local
112 const accscalar_t divide_factor = static_cast<accscalar_t>(kT * kH * kW); in adaptiveaveragepool() local205 const accscalar_t divide_factor = kW * kH * kT; in adaptiveaveragegradinput() local
216 int64_t divide_factor = 0; in avg_pool3d_out_frame() local393 int64_t divide_factor = 0; in avg_pool3d_backward_out_frame() local
86 int64_t divide_factor; in avg_pool2d_out_frame() local
2058 int divide_factor = local