@testing.for_all_dtypes(no_complex=True)
@testing.numpy_cupy_allclose()
def test_argmax_all(self, xp, dtype):
    a = testing.shaped_random((2, 3), xp, dtype)
    return a.argmax()
@testing.for_all_dtypes(no_complex=True)
@testing.numpy_cupy_allclose()
def test_argmax_all(self, xp, dtype):
    a = testing.shaped_random((2, 3), xp, dtype)
    return a.argmax()