Home
last modified time | relevance | path

Searched defs:pwend (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op_gpu.cu.cc59 const int pwend = min(w / stride_w + 1, pooled_width); in DEFINE_GPU_KERNELS() local
/external/pytorch/aten/src/ATen/native/cuda/
DAveragePool2d.cu148 const int pwend = min(w / stride_w + 1, pooled_width); in avg_pool2d_backward_out_cuda_frame() local
204 const int pwend = min(w / stride_w + 1, pooled_width); in avg_pool2d_backward_out_cuda_frame_nhwc() local
DDilatedMaxPool2d.cu190 int pwend = p_end(w, pad_w, pooled_width, stride_w); in max_pool_backward_nchw() local
253 int pwend = p_end(iw, pad_w, pooled_width, stride_w); in C10_LAUNCH_BOUNDS_1() local