What type of PR is this?

Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line:

/kind bug

What does this PR do / why do we need it:
fix argmaxwithvalue error in pynative mode on GPU

Which issue(s) this PR fixes:

Fixes ##I1NVX0:argmaxwithvalue run failed in pynative mode on GPU

Special notes for your reviewers: