PR types

Bug fixes

PR changes

OPs

Describe

Fix dtype for arg_max op and also fix dygraph op of softmax_with_cross_entroy.