Searched refs:logit_backward_kernel (Results 1 – 1 of 1) sorted by relevance
856 void logit_backward_kernel(TensorIteratorBase& iter, const Scalar& eps_scalar) { in logit_backward_kernel() function1429 ALSO_REGISTER_AVX512_DISPATCH(logit_backward_stub, &logit_backward_kernel);