Searched refs:batch_norm_collect_statistics_kernel (Results 1 – 1 of 1) sorted by relevance
269 __global__ void batch_norm_collect_statistics_kernel( in batch_norm_collect_statistics_kernel() function688 …batch_norm_collect_statistics_kernel<VarTransform, scalar_t, scalar_t, accscalar_t, index_t> <<<bl… in batch_norm_stats_cuda_template()