Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAveragePool2d.cu145 const int phstart = (h < kernel_h) ? 0 : (h - kernel_h) / stride_h + 1; in avg_pool2d_backward_out_cuda_frame() local
201 const int phstart = (h < kernel_h) ? 0 : (h - kernel_h) / stride_h + 1; in avg_pool2d_backward_out_cuda_frame_nhwc() local
H A DDilatedMaxPool2d.cu187 int phstart = p_start(h, pad_h, kernel_h, dilation_h, stride_h); in max_pool_backward_nchw() local
249 int phstart = p_start(ih, pad_h, kernel_h, dilation_h, stride_h); in C10_LAUNCH_BOUNDS_1() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Davgpooling_op_gpu.cu.cc56 const int phstart = (h < kernel_h) ? 0 : (h - kernel_h) / stride_h + 1; in DEFINE_GPU_KERNELS() local