-
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][TENSORFLOW] Optimized tensorflow testcases #1546
Conversation
Suggest to wait a while until #1505 is merged. |
please resync the changes against current master as in #1394 |
@@ -82,6 +82,34 @@ def run_tf_graph(sess, input_data, input_node, output_node): | |||
output_data = sess.run(tensor, input_dict) | |||
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
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.
@@ -1007,11 +822,11 @@ def test_forward_l2_normalize(): | |||
_test_forward_concat_v2() | |||
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? |
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