Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cuh172 const T& m2n_new) { in welford_merge_element()
204 auto m2n_new = shmem_m2n[address]; in welford_merge_block_vertical() local
1062 accscalar_t m2n_new = c_offset < stride ? staging_m2n[address_base] : accscalar_t(0.0); in batch_norm_collect_statistics_channels_last_kernel() local