Searched refs:log_softmax_kernel_impl (Results 1 – 1 of 1) sorted by relevance
1230 static void log_softmax_kernel_impl(const Tensor& result, const Tensor& self, int64_t dim) { in log_softmax_kernel_impl() function1304 ALSO_REGISTER_AVX512_DISPATCH(log_softmax_kernel, &log_softmax_kernel_impl);