Searched refs:argmax_kernel_cuda (Results 1 – 1 of 1) sorted by relevance
29 void argmax_kernel_cuda(TensorIterator& iter) { in argmax_kernel_cuda() function44 REGISTER_DISPATCH(argmax_stub, &argmax_kernel_cuda);