Searched refs:logit_kernel_cuda (Results 1 – 1 of 1) sorted by relevance
193 void logit_kernel_cuda(TensorIteratorBase& iter, const Scalar& eps_scalar) { in logit_kernel_cuda() function388 REGISTER_DISPATCH(logit_stub, &logit_kernel_cuda);