Searched refs:batch_norm_transform_input_channels_last_kernel (Results 1 – 1 of 1) sorted by relevance
1088 __global__ void batch_norm_transform_input_channels_last_kernel( in batch_norm_transform_input_channels_last_kernel() function1489 …batch_norm_transform_input_channels_last_kernel<scalar_t, accscalar_t, accscalar_t, ELEMENTS_PER_I… in batch_norm_elemt_channels_last_cuda_template()1510 …batch_norm_transform_input_channels_last_kernel<scalar_t, accscalar_t, scalar_t, ELEMENTS_PER_ITER> in batch_norm_elemt_channels_last_cuda_template()