Searched refs:batch_norm_collect_statistics_channels_last_kernel (Results 1 – 1 of 1) sorted by relevance
944 batch_norm_collect_statistics_channels_last_kernel( in batch_norm_collect_statistics_channels_last_kernel() function1453 …batch_norm_collect_statistics_channels_last_kernel<VarTransform, scalar_t, accscalar_t, ELEMENTS_P… in batch_norm_stats_channels_last_cuda_template()