Searched refs:batch_norm_transform_input_kernel (Results 1 – 1 of 1) sorted by relevance
212 __global__ void batch_norm_transform_input_kernel( in batch_norm_transform_input_kernel() function730 …batch_norm_transform_input_kernel<input_scalar_t, stat_scalar_t, stat_accscalar_t, true, index_t> … in batch_norm_elemt_cuda_template()