Searched refs:batch_norm_backward_elemt_kernel (Results 1 – 1 of 1) sorted by relevance
546 __global__ void batch_norm_backward_elemt_kernel( in batch_norm_backward_elemt_kernel() function568 __global__ void batch_norm_backward_elemt_kernel( in batch_norm_backward_elemt_kernel() function880 batch_norm_backward_elemt_kernel<input_scalar_t, stat_scalar_t, stat_accscalar_t, index_t> in batch_norm_backward_elemt_cuda_template()929 …batch_norm_backward_elemt_kernel<input_scalar_t, stat_scalar_t, stat_accscalar_t, index_t> <<<bloc… in batch_norm_backward_elemt_cuda_template()