-
Notifications
You must be signed in to change notification settings - Fork 3.8k
[FRONTEND][TENSORFLOW] Optimized tensorflow testcases #1546
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
Conversation
|
Suggest to wait a while until #1505 is merged. |
|
please resync the changes against current master as in #1394 |
| return output_data | ||
|
|
||
|
|
||
| def compate_tf_with_tvm(in_data, in_name, out_name, init_global_variables=False): |
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.
typo compate
| [tf_out[0].shape, (2, batch_size, num_hidden)], | ||
| [tf_out[0].dtype, tf_out[1].dtype]) | ||
|
|
||
| if isinstance(tvm_output, list): |
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.
assert tvm_output is list?
|
@srkreddy1238 @yzhliu I have handled the review comments and rebased the PR, |
1aad10c to
0ecd582
Compare
srkreddy1238
left a comment
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.
Thanks @joyalbin for converting Constants to Placeholders.
Please handle the comment, otherwise LGTM.
| test_forward_multi_input() | ||
| test_forward_inception_v3() | ||
| test_forward_inception_v1() | ||
| #test_forward_inception_v1() |
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.
Pls remove local changes.
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.
done
|
@srkreddy1238 would you mind take another look and approve if it is good to you? |
srkreddy1238
left a comment
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 after rebase and CI pass.
|
Thanks, this is now merged. |
* [NNVM][TENSORFLOW] Optimized tensorflow testcases * Replace Constants with Placeholder * Review comment fix
This PR is to remove the code duplication in tensoflow testcases