Created by: wawltor

PR types

Others

PR changes

OPs

Describe

fix the argmin,argmax op for the paddlepaddle 2.0， cherry-pick from PR 26792

• Registered the AddCheckPoint for the argmax, argmin, add the flatten attribuate for the argmax/argmin, the flatten means that flattening the tensor as input;At the same time, change the default value from -1 to 3 in the attribute of argmax/argmin, but do not change the operator logic.

• Fixed the output of argmax/argmin, because the paddlepaddle do not support the scalar tensor, change the output of argmin/argmax from the scalar tensor to normal tensor with shape [1] .

• Added the check for dtype of argmin/argmax, when the num of element in argmax/argmin is larger the max value of int32, remind the user to use the dtype of int64.