Searched refs:softmax_lastdim_kernel_impl (Results 1 – 1 of 1) sorted by relevance
1206 static void softmax_lastdim_kernel_impl( in softmax_lastdim_kernel_impl() function1294 ALSO_REGISTER_AVX512_DISPATCH(softmax_lastdim_kernel, &softmax_lastdim_kernel_impl);