Searched defs:o_avg (Results 1 – 1 of 1) sorted by relevance
307 stat_accscalar_t o_avg = WARP_SHFL_XOR(avg, 1 << i, C10_WARP_SIZE); in batch_norm_collect_statistics_kernel() local335 stat_accscalar_t o_avg = WARP_SHFL_XOR(avg, 1 << i, C10_WARP_SIZE); in batch_norm_collect_statistics_kernel() local