Searched refs:logit_backward_kernel_cuda (Results 1 – 1 of 1) sorted by relevance
56 void logit_backward_kernel_cuda(TensorIteratorBase& iter, const Scalar& eps_scalar) { in logit_backward_kernel_cuda() function128 REGISTER_DISPATCH(logit_backward_stub, &logit_backward_kernel_cuda);