Searched refs:topk_impl_loop (Results 1 – 3 of 3) sorted by relevance
252 return topk_impl_loop<scalar_t, float>( in topk_kernel()256 return topk_impl_loop<scalar_t, scalar_t>( in topk_kernel()
15 void topk_impl_loop( in topk_impl_loop() function
2393 return topk_impl_loop<underlying_t, underlying_t>(