Searched refs:batch_norm_cpu_collect_stats_contiguous_internal (Results 1 – 1 of 1) sorted by relevance
830 inline void batch_norm_cpu_collect_stats_contiguous_internal( in batch_norm_cpu_collect_stats_contiguous_internal() function896 …batch_norm_cpu_collect_stats_contiguous_internal<scalar_t, at::opmath_type<scalar_t>>(mean, var_su… in batch_norm_cpu_collect_stats_contiguous_impl()898 batch_norm_cpu_collect_stats_contiguous_internal<scalar_t, scalar_t>(mean, var_sum, input); in batch_norm_cpu_collect_stats_contiguous_impl()