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

[FRONTEND][TFLITE] Add FULLY_CONNECTED op into tflite frontend, support Inception V4 #3019

Merged
merged 2 commits into from
May 1, 2019

Conversation

songqun
Copy link
Contributor

@songqun songqun commented Apr 13, 2019

add FULLY_CONNECTED op into tflite frontend to support Inception V4.
see https://discuss.tvm.ai/t/tflite-fully-connected/2192

@songqun songqun changed the title Add FULLY_CONNECTED op into tflite frontend, support Inception V4 [FRONTEND][TFLITE]Add FULLY_CONNECTED op into tflite frontend, support Inception V4 Apr 13, 2019
@songqun songqun changed the title [FRONTEND][TFLITE]Add FULLY_CONNECTED op into tflite frontend, support Inception V4 [FRONTEND][TFLITE] Add FULLY_CONNECTED op into tflite frontend, support Inception V4 Apr 13, 2019
Copy link
Member

@FrozenGene FrozenGene left a comment

Choose a reason for hiding this comment

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

LGTM.

@@ -512,6 +563,21 @@ def test_forward_inception_v3_net():
tvm.testing.assert_allclose(np.squeeze(tvm_output[0]), np.squeeze(tflite_output[0]),
rtol=1e-5, atol=1e-5)

def test_forward_inception_v4_net():
'''test inception v4 tflite model'''
Copy link
Member

Choose a reason for hiding this comment

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

Suggested change
'''test inception v4 tflite model'''
"""Test the Inception v4 TF Lite model."""

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Do I need to fix the comment style for all other end-to-end tests such as """Test the Mobilenet V1 TF Lite model.""" instead of '''test mobilenet v1 tflite model'''?

@jroesch
Copy link
Member

jroesch commented Apr 15, 2019

Can you just fix the message and then I will merge 😄

@tqchen tqchen added the status: need update need update based on feedbacks label Apr 16, 2019
@tqchen
Copy link
Member

tqchen commented Apr 20, 2019

ping @jroesch

@songqun
Copy link
Contributor Author

songqun commented Apr 25, 2019

@jroesch Could you help me merge the branch?

@jroesch
Copy link
Member

jroesch commented May 1, 2019

Sorry about the slow response, been very busy :)

@jroesch jroesch merged commit 78e0871 into apache:master May 1, 2019
wweic pushed a commit to wweic/tvm that referenced this pull request May 13, 2019
…rt Inception V4 (apache#3019)

* Add FULLY_CONNECTED op into tflite frontend, support Inception V4

* Fix comment style in TF Lite tests.
wweic pushed a commit to neo-ai/tvm that referenced this pull request May 13, 2019
…rt Inception V4 (apache#3019)

* Add FULLY_CONNECTED op into tflite frontend, support Inception V4

* Fix comment style in TF Lite tests.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
status: need update need update based on feedbacks
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants