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

[TFLite] Convert TFLite NCHW to NHWC #3141

Open
wants to merge 2 commits into
base: master
from

Conversation

Projects
None yet
3 participants
@FrozenGene
Copy link
Contributor

commented May 7, 2019

Background RFC: #2519

As discussed in above RFC, we agree to make TFLite frontend input layout from NCHW to NHWC. This PR does this.

Affected:
We could not use Auto TVM to tuning on ARM CPU. Because ARM CPU schedule only implement NCHW currently. We should add SpatialPack + NHWC for conv2d / depthwise convolution on ARM CPU.

@srkreddy1238

cc: @songqun @gomida @ariwaranosai @tqchen @yzhliu

@FrozenGene FrozenGene force-pushed the FrozenGene:tflite_nhwc branch from ba85c70 to ee721de May 7, 2019

@FrozenGene

This comment has been minimized.

Copy link
Contributor Author

commented May 14, 2019

@srkreddy1238

This comment has been minimized.

Copy link
Member

commented May 15, 2019

@FrozenGene overall look good to me.

Trigger the CI and I will have another look.

@FrozenGene FrozenGene force-pushed the FrozenGene:tflite_nhwc branch from ee721de to 50b9138 May 15, 2019

@FrozenGene

This comment has been minimized.

Copy link
Contributor Author

commented May 16, 2019

@srkreddy1238 CI is green now.

@FrozenGene

This comment has been minimized.

Copy link
Contributor Author

commented May 20, 2019

@srkreddy1238 Could you help to have a look now? Thanks.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.