Searched defs:shmem_mean (Results 1 – 1 of 1) sorted by relevance
186 T* shmem_mean, in welford_merge_block_vertical()1021 static __shared__ accscalar_t shmem_mean[MAX_BLOCK_SIZE]; in batch_norm_collect_statistics_channels_last_kernel() local