Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
DNormalization.cuh170 const C& count_new, in welford_merge_element()
202 auto count_new = shmem_count[address]; in welford_merge_block_vertical() local
1060 int count_new = c_offset < stride ? staging_count[address_base] : 0; in batch_norm_collect_statistics_channels_last_kernel() local