Skip to content

[ARM]opt argmax op,test=develop#9384

Merged
mjp9527 merged 1 commit intoPaddlePaddle:developfrom
wz1qqx:opt-argmax
Sep 16, 2022
Merged

[ARM]opt argmax op,test=develop#9384
mjp9527 merged 1 commit intoPaddlePaddle:developfrom
wz1qqx:opt-argmax

Conversation

@wz1qqx
Copy link
Collaborator

@wz1qqx wz1qqx commented Aug 29, 2022

PR types

[Others ]

PR changes

[ OPs]

Describe

修改了argmax算子的实现,采取遍历的方式,较原有partitial_sort方法性能提升一倍。

@paddle-bot
Copy link

paddle-bot bot commented Aug 29, 2022

Thanks for your contribution!

@CLAassistant
Copy link

CLAassistant commented Aug 29, 2022

CLA assistant check
All committers have signed the CLA.

@wz1qqx wz1qqx changed the title opt argmax op,test=develop [ARM]opt argmax op,test=develop Aug 29, 2022
Copy link
Contributor

@liyancas liyancas left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@wz1qqx wz1qqx closed this Sep 15, 2022
@wz1qqx wz1qqx reopened this Sep 15, 2022
@mjp9527 mjp9527 merged commit 125caa1 into PaddlePaddle:develop Sep 16, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants