Searched refs:vec_host_softmax_lastdim (Results 1 – 1 of 1) sorted by relevance
671 struct vec_host_softmax_lastdim { struct1212 [&] { vec_host_softmax_lastdim<scalar_t, false>::apply(result, self); }); in softmax_lastdim_kernel_impl()1227 [&] { vec_host_softmax_lastdim<scalar_t, true>::apply(result, self); }); in log_softmax_lastdim_kernel_impl()