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

[TensorFlow] Fix limitation that depth_mult can only be 1 for DepthwiseConv2dNative #3676

Merged
merged 2 commits into from
Aug 29, 2019

Conversation

lixiaoquan
Copy link
Contributor

@lixiaoquan lixiaoquan commented Jul 31, 2019

@srkreddy1238 @yongwww Please review this

@tqchen
Copy link
Member

tqchen commented Aug 1, 2019

@srkreddy1238 please help manage the PR

@FrozenGene
Copy link
Member

Affected : https://discuss.tvm.ai/t/deployment-of-converted-pb-file-to-tflite-using-relay-frontend-from-tflite/3307/13

When this PR merged, I could support tflite frontend.

@tqchen
Copy link
Member

tqchen commented Aug 12, 2019

ping @srkreddy1238 , @kazum can you also please help to take a look?

Copy link
Contributor

@kazum kazum left a comment

Choose a reason for hiding this comment

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

Overall, it looks good to me. I added some comments for better code readability.

@@ -129,19 +129,20 @@ def compute_conv2d(attrs, inputs, out_type, target):
if dilation_h < 1 or dilation_w < 1:
raise ValueError("dilation should be positive value")

def _get_channels():
Copy link
Contributor

Choose a reason for hiding this comment

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

Is _get_out_depth() better because the returned value makes sense only for depthwise convolution?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

updated

src/relay/op/nn/convolution.cc Show resolved Hide resolved
@lixiaoquan
Copy link
Contributor Author

@tqchen Could you help to merge this? Thanks

@tqchen tqchen merged commit ce03143 into apache:master Aug 29, 2019
@tqchen
Copy link
Member

tqchen commented Aug 29, 2019

Thanks @lixiaoquan @kazum !

wweic pushed a commit to wweic/tvm that referenced this pull request Sep 16, 2019
…seConv2dNative (apache#3676)

* [TensorFlow] Fix limitation that depth_mult can only be 1 for DepthwiseConv2dNative

* Improve code readability
wweic pushed a commit to wweic/tvm that referenced this pull request Sep 16, 2019
…seConv2dNative (apache#3676)

* [TensorFlow] Fix limitation that depth_mult can only be 1 for DepthwiseConv2dNative

* Improve code readability
wweic pushed a commit to neo-ai/tvm that referenced this pull request Sep 16, 2019
…seConv2dNative (apache#3676)

* [TensorFlow] Fix limitation that depth_mult can only be 1 for DepthwiseConv2dNative

* Improve code readability
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.

5 participants