Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
Dlayer_norm_kernel.cu179 WelfordDataLN wdB{WARP_SHFL_DOWN(wd.mean, offset), in compute_stats() local
199 WelfordDataLN wdB{meansigmabuf[2*threadIdx.y], in compute_stats() local