Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DNormalization.cuh307 stat_accscalar_t o_avg = WARP_SHFL_XOR(avg, 1 << i, C10_WARP_SIZE); in batch_norm_collect_statistics_kernel() local
335 stat_accscalar_t o_avg = WARP_SHFL_XOR(avg, 1 << i, C10_WARP_SIZE); in batch_norm_collect_statistics_kernel() local