Searched refs:logit_kernel (Results 1 – 1 of 1) sorted by relevance
140 static void logit_kernel(TensorIteratorBase& iter, const Scalar& eps_scalar) { in logit_kernel() function848 ALSO_REGISTER_AVX512_DISPATCH(logit_stub, &CPU_CAPABILITY::logit_kernel);