-
Notifications
You must be signed in to change notification settings - Fork 3.8k
Add Split and realdiv op support #2123
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
|
For split, my pull request covers |
|
Thanks for contributing, please request review from Reviewers |
|
|
||
| pad_v = _get_pad_pair(in_h, kernel_h, stride_h) | ||
| pad_h = _get_pad_pair(in_w, kernel_w, stride_w) | ||
| dilation_h = attr['dilations'][0] |
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.
@Rasterer Can you explain this change ?
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.
Please refer to https://github.com/dmlc/tvm/blob/master/nnvm/src/top/nn/convolution.cc#L100 for the affect of dilation on output size. Dilated conv always can be simulated by a normal conv with dilated kernel. So I use dilated kernel size for pad calculation instead.
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.
@srkreddy1238 any more comment about this?
|
@srkreddy1238 could you verify the changes? |
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 @Rasterer. sorry about the delay, few comments before concluding the PR.
|
|
||
| ####################################################################### | ||
| # Split | ||
| # ------ |
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.
Under line up to the string pls.
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.
@srkreddy1238 updated
| # Split | ||
| # ------ | ||
|
|
||
| def _test_split(in_shape, axis, num_split, dtype): |
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.
Can you add another test case with split followed by concat ?
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.
@srkreddy1238 updated
* Add Split and realdiv op support * Fix the pad calculation in the case of dilated convolution
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 @Rasterer LGTM.
|
cc @yzhliu |
|
Thanks @Rasterer @srkreddy1238 . This is now merged. |
* Add Split and realdiv op support * Fix the pad calculation in the case of dilated convolution
* Add Split and realdiv op support * Fix the pad calculation in the case of dilated convolution
* Add Split and realdiv op support * Fix the pad calculation in the case of dilated convolution
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 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
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 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
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 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
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
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
* [FRONTEND][TENSORFLOW] Enhance with left over patches from NNVM. 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 * * review comments * * resnet fix. * * review comments
…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
…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
Add Split and realdiv op support
Fix the pad calculation in the case of dilated convolution
@srkreddy1238 @nishi-t please help review, thanks