Searched refs:argmax_kernel_impl (Results 1 – 1 of 1) sorted by relevance
388 static void argmax_kernel_impl(TensorIterator &iter) { in argmax_kernel_impl() function448 REGISTER_DISPATCH(argmax_stub, &argmax_kernel_impl);