Searched refs:batch_norm_backward_kernel (Results 1 – 1 of 1) sorted by relevance
356 __global__ void batch_norm_backward_kernel( in batch_norm_backward_kernel() function652 …batch_norm_backward_kernel<input_scalar_t, stat_scalar_t, accscalar_t, index_t> <<<blocks, threads… in batch_norm_backward_cuda_template()