IVGCVSW-3944 Add ArgMinMax output shape validation