Searched refs:vectorized_layer_norm_kernel_impl (Results 1 – 1 of 1) sorted by relevance
221 __device__ __inline__ void vectorized_layer_norm_kernel_impl( in vectorized_layer_norm_kernel_impl() function286 __device__ __inline__ void vectorized_layer_norm_kernel_impl( in vectorized_layer_norm_kernel_impl() function309 vectorized_layer_norm_kernel_impl(N, eps, X, gamma, beta, mean, rstd, Y); in vectorized_layer_norm_kernel()