代码拉取完成,页面将自动刷新
2229
[NPU] fix dtype for arg_max, test=develop
已合并
PR types
Bug fixes
PR changes
OPs
Describe
Fix dtype for arg_max op and also fix dygraph op of softmax_with_cross_entroy.
Bug fixes
OPs
Fix dtype for arg_max op and also fix dygraph op of softmax_with_cross_entroy.