Searched refs:batch_norm_cpu_collect_stats_contiguous_impl (Results 1 – 1 of 1) sorted by relevance
177 batch_norm_cpu_collect_stats_contiguous_impl( in batch_norm_cpu_collect_stats_contiguous_impl() function892 batch_norm_cpu_collect_stats_contiguous_impl( in batch_norm_cpu_collect_stats_contiguous_impl() function1360 batch_norm_cpu_collect_stats_contiguous_impl<scalar_t>(mean, var_sum, input); in batch_norm_cpu_collect_stats_kernel()