Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Improve x86 roi align #3296

Merged
merged 2 commits into from
Jun 6, 2019
Merged

Improve x86 roi align #3296

merged 2 commits into from
Jun 6, 2019

Conversation

kevinthesun
Copy link
Contributor

Add x86 implementation for roi_align_nchw with hybrid script to improve performance for cpu.

@kevinthesun kevinthesun requested a review from vinx13 June 5, 2019 22:38
@vinx13 vinx13 self-assigned this Jun 6, 2019
rounded_div_h = int32(div_h) * 1.0
rounded_div_w = int32(div_w) * 1.0
if sample_ratio <= 0:
# Cannot use ceil function since hybrid script
Copy link
Member

Choose a reason for hiding this comment

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

cc @were

@vinx13 vinx13 merged commit 9164809 into apache:master Jun 6, 2019
@kevinthesun kevinthesun deleted the ImproveX86ROIAlign branch June 13, 2019 20:10
wweic pushed a commit to wweic/tvm that referenced this pull request Jun 26, 2019
* Improve roi_align performance for x86

* Change test
wweic pushed a commit to neo-ai/tvm that referenced this pull request Jun 27, 2019
* Improve roi_align performance for x86

* Change test
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants