Skip to content

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