-
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] Enhance with left over patches from NNVM. #2757
Conversation
Also @yongwww & @ashutoshparkhi welcome to review |
commit 76188a4 Author: Siva sivar.b@huawei.com [NNVM][TENSORFLOW] bugfix. (apache#2444) commit 6737739 Author: Ashutosh Parkhi ashutosh.parkhi@imgtec.com [Tensorflow] Support for Crop (apache#2285) commit f6c3f99 Author: Alexey Romanov alexey.v.romanov@gmail.com [FRONTEND][TENSORFLOW] Use input shapes directly instead of 1-element lists (apache#2242) commit e5d92e1 Author: Dominic Symes 36929632+dominicsymes@users.noreply.github.com [FRONTEND][TENSORFLOW] Bugfix (apache#2326) commit 00d509d Author: Alexey Romanov alexey.v.romanov@gmail.com [FRONTEND][TENSORFLOW] Support Unstack and Split (apache#2105) commit df9d3ad Author: Siva sivar.b@huawei.com [FRONTEND][TENSORFLOW] Bugfix (apache#2267) commit d1a0c90 Author: Zhebin Jin zhebin.jzb@alibaba-inc.com [FRONTEND][TENSORFLOW]Add Split and realdiv op support (apache#2123) * Add Split and realdiv op support * Fix the pad calculation in the case of dilated convolution
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.
Looks good to me, thanks!
@@ -1267,15 +1369,22 @@ def from_tensorflow(self, graph, layout="NHWC", shape=None, outputs=None): | |||
elif shape and node.name in shape: | |||
# Give priority to user argument. | |||
self._output_shapes[node.name] = [shape[node.name]] | |||
elif node.op == 'Placeholder': |
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.
Actually the change here was merged with pr #2493. It is okay to ignore the following three commits since #2493 was merged.
commit f347b52
Author: Yong Wu yongwu@amazon.com
Get tags of saved model automatically
commit 916576c
Author: Zhi Chen chzhi@amazon.com
Support TensorFlow saved model
TF parser: return the consistent error message to error handler
commit f1782f3
Author: Yong Wu yongwu@amazon.com
Add tf parser wrapper, infer shape automatically
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.
You are right. Removed now.
@srkreddy1238 please send another push to re-trigger CI. @yongwww please https://docs.tvm.ai/contribute/code_review.html#approve-and-request-changes-explicitly |
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 for the work!
…che#2757) * [FRONTEND][TENSORFLOW] Enhance with left over patches from NNVM. commit 76188a4 Author: Siva sivar.b@huawei.com [NNVM][TENSORFLOW] bugfix. (apache#2444) commit 6737739 Author: Ashutosh Parkhi ashutosh.parkhi@imgtec.com [Tensorflow] Support for Crop (apache#2285) commit f6c3f99 Author: Alexey Romanov alexey.v.romanov@gmail.com [FRONTEND][TENSORFLOW] Use input shapes directly instead of 1-element lists (apache#2242) commit e5d92e1 Author: Dominic Symes 36929632+dominicsymes@users.noreply.github.com [FRONTEND][TENSORFLOW] Bugfix (apache#2326) commit 00d509d Author: Alexey Romanov alexey.v.romanov@gmail.com [FRONTEND][TENSORFLOW] Support Unstack and Split (apache#2105) commit df9d3ad Author: Siva sivar.b@huawei.com [FRONTEND][TENSORFLOW] Bugfix (apache#2267) commit d1a0c90 Author: Zhebin Jin zhebin.jzb@alibaba-inc.com [FRONTEND][TENSORFLOW]Add Split and realdiv op support (apache#2123) * Add Split and realdiv op support * Fix the pad calculation in the case of dilated convolution * * review comments * * resnet fix. * * review comments
# This means the node is 1d in Relay and 0d in TF. | ||
# See `_expand_dims_0d_aware`. | ||
if self._outputs_are_0d[node_name][tensor_slot] and input_shape: | ||
input_0d_mismatch.add(in_sym) |
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.
Shouldn't this be in_sym[tensor_slot]?
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.
I think tensor_slot us already handled above.
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.
In my variant of inception_v3, in_sym comes as a list of Var instead of _expr.TupleWrapper.
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.
@ashutoshparkhi will fix it in pr #2830
…che#2757) * [FRONTEND][TENSORFLOW] Enhance with left over patches from NNVM. commit 76188a4 Author: Siva sivar.b@huawei.com [NNVM][TENSORFLOW] bugfix. (apache#2444) commit 6737739 Author: Ashutosh Parkhi ashutosh.parkhi@imgtec.com [Tensorflow] Support for Crop (apache#2285) commit f6c3f99 Author: Alexey Romanov alexey.v.romanov@gmail.com [FRONTEND][TENSORFLOW] Use input shapes directly instead of 1-element lists (apache#2242) commit e5d92e1 Author: Dominic Symes 36929632+dominicsymes@users.noreply.github.com [FRONTEND][TENSORFLOW] Bugfix (apache#2326) commit 00d509d Author: Alexey Romanov alexey.v.romanov@gmail.com [FRONTEND][TENSORFLOW] Support Unstack and Split (apache#2105) commit df9d3ad Author: Siva sivar.b@huawei.com [FRONTEND][TENSORFLOW] Bugfix (apache#2267) commit d1a0c90 Author: Zhebin Jin zhebin.jzb@alibaba-inc.com [FRONTEND][TENSORFLOW]Add Split and realdiv op support (apache#2123) * Add Split and realdiv op support * Fix the pad calculation in the case of dilated convolution * * review comments * * resnet fix. * * review comments
commit 76188a4
Author: Siva sivar.b@huawei.com
[NNVM][TENSORFLOW] bugfix. (#2444)
commit 6737739
Author: Ashutosh Parkhi ashutosh.parkhi@imgtec.com
[Tensorflow] Support for Crop (#2285)
commit f6c3f99
Author: Alexey Romanov alexey.v.romanov@gmail.com
[FRONTEND][TENSORFLOW] Use input shapes directly instead of 1-element lists (#2242)
commit e5d92e1
Author: Dominic Symes 36929632+dominicsymes@users.noreply.github.com
[FRONTEND][TENSORFLOW] Bugfix (#2326)
commit 00d509d
Author: Alexey Romanov alexey.v.romanov@gmail.com
[FRONTEND][TENSORFLOW] Support Unstack and Split (#2105)
commit df9d3ad
Author: Siva sivar.b@huawei.com
[FRONTEND][TENSORFLOW] Bugfix (#2267)
commit d1a0c90
Author: Zhebin Jin zhebin.jzb@alibaba-inc.com
[FRONTEND][TENSORFLOW]Add Split and realdiv op support (#2123)
* Add Split and realdiv op support
* Fix the pad calculation in the case of dilated convolution
Thanks for contributing to TVM! Please refer to guideline https://docs.tvm.ai/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers.