Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DNormalization.cuh308 int o_n = WARP_SHFL_XOR(n, 1 << i, C10_WARP_SIZE); in batch_norm_collect_statistics_kernel() local
336 int o_n = WARP_SHFL_XOR(n, 1 << i, C10_WARP_SIZE); in batch_norm_collect_statistics_kernel() local