Searched refs:LayerNormForwardCUDAKernel (Results 1 – 1 of 1) sorted by relevance
93 __global__ void LayerNormForwardCUDAKernel( in LayerNormForwardCUDAKernel() function797 LayerNormForwardCUDAKernel<T, T_ACC><<<M, kCUDANumThreads, 0, cuda_stream>>>( in LayerNormKernelImplInternal()