-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
Conversation
There was a problem hiding this 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''' |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
'''test inception v4 tflite model''' | |
"""Test the Inception v4 TF Lite model.""" |
There was a problem hiding this comment.
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'''?
Can you just fix the message and then I will merge 😄 |
ping @jroesch |
@jroesch Could you help me merge the branch? |
Sorry about the slow response, been very busy :) |
…rt Inception V4 (apache#3019) * Add FULLY_CONNECTED op into tflite frontend, support Inception V4 * Fix comment style in TF Lite tests.
…rt Inception V4 (apache#3019) * Add FULLY_CONNECTED op into tflite frontend, support Inception V4 * Fix comment style in TF Lite tests.
add FULLY_CONNECTED op into tflite frontend to support Inception V4.
see https://discuss.tvm.ai/t/tflite-fully-connected/2192