Searched refs:softmax_kernel_impl (Results 1 – 1 of 1) sorted by relevance
1215 static void softmax_kernel_impl(const Tensor& result, const Tensor& self, int64_t dim) { in softmax_kernel_impl() function1303 ALSO_REGISTER_AVX512_DISPATCH(softmax_kernel, &softmax_kernel_impl);