Searched refs:batch_norm_reduce_statistics_kernel (Results 1 – 1 of 1) sorted by relevance
427 __global__ void batch_norm_reduce_statistics_kernel( in batch_norm_reduce_statistics_kernel() function770 batch_norm_reduce_statistics_kernel<scalar_t, accscalar_t, index_t> <<<grid, block, 0, stream>>> in batch_norm_gather_stats_cuda_template()