Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAveragePool2d.cu147 const int pwstart = (w < kernel_w) ? 0 : (w - kernel_w) / stride_w + 1; in avg_pool2d_backward_out_cuda_frame() local
203 const int pwstart = (w < kernel_w) ? 0 : (w - kernel_w) / stride_w + 1; in avg_pool2d_backward_out_cuda_frame_nhwc() local
H A DDilatedMaxPool2d.cu189 int pwstart = p_start(w, pad_w, kernel_w, dilation_w, stride_w); in max_pool_backward_nchw() local
252 int pwstart = p_start(iw, pad_w, kernel_w, dilation_w, stride_w); in C10_LAUNCH_BOUNDS_1() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Davgpooling_op_gpu.cu.cc58 const int pwstart = (w < kernel_w) ? 0 : (w - kernel_w) / stride_w + 1; in DEFINE_GPU_KERNELS() local