Searched refs:wend (Results 1 – 4 of 4) sorted by relevance
80 int wend = min(wstart + kernel_w, width); in MaxPoolForwardNCHW() local87 for (int w = wstart; w < wend; ++w) { in MaxPoolForwardNCHW()123 int wend = min(wstart + kernel_w, width); in MaxPoolForwardNoMaskKernel_NCHW_VECT_C() local129 for (int w = wstart; w < wend; ++w) { in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()156 int wend = min(wstart + kernel_w, width); in MaxPoolForwardNHWC() local163 for (int w = wstart; w < wend; ++w) { in MaxPoolForwardNHWC()195 int wend = min(wstart + kernel_w, width); in MaxPoolBackwardNoMaskNHWC() local202 for (int w = wstart; w < wend; ++w) { in MaxPoolBackwardNoMaskNHWC()278 const int wend = min(wstart + kernel_w, width); in MaxPoolGradBackwardNoMaskNCHW() local286 for (int w = wstart; w < wend && !should_stop; ++w) { in MaxPoolGradBackwardNoMaskNCHW()[all …]
50 const int wend = min(wstart + kernel_w, width); in MaxPoolGradBackwardNoMaskNCDHW() local61 for (int w = wstart; w < wend && !should_stop; ++w) { in MaxPoolGradBackwardNoMaskNCDHW()94 int wend = min(wstart + kernel_w, width); in MaxPoolGradBackwardNoMaskNDHWC() local112 for (int w = wstart; w < wend && !should_stop; ++w) { in MaxPoolGradBackwardNoMaskNDHWC()
70 int wend = min(wstart + kernel_w, width); in DEFINE_GPU_KERNELS() local73 int pool_size = (hend - hstart) * (wend - wstart); in DEFINE_GPU_KERNELS()
523 int wend; // The bottom bit of the window in mod_exp_recp() local543 wend = 0; in mod_exp_recp()549 wvalue <<= (i - wend); in mod_exp_recp()551 wend = i; in mod_exp_recp()556 j = wend + 1; in mod_exp_recp()572 wstart -= wend + 1; in mod_exp_recp()