Searched refs:vec_host_softmax_backward_lastdim (Results 1 – 1 of 1) sorted by relevance
1148 struct vec_host_softmax_backward_lastdim { struct1243 vec_host_softmax_backward_lastdim<scalar_t, false>::apply( in softmax_backward_lastdim_kernel_impl()1255 vec_host_softmax_backward_lastdim<scalar_t, true>::apply( in log_softmax_backward_lastdim_kernel_impl()