Skip to content
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][Tensorflow2] Tf2 tensorlist support #215

Conversation

zxy844288792
Copy link

@zxy844288792 zxy844288792 commented Jul 15, 2021

This pr includes several pr related to tensorlist changes

https://quip-amazon.com/ZIZ7AzO1pZbL/Neo-TVM-110-Update
it includes
apache#8238
apache#8464
apache#8558

Thanks for contributing to TVM! Please refer to guideline https://tvm.apache.org/docs/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers by @ them in the pull request thread.

Xingyu Zhou and others added 3 commits July 15, 2021 18:15
add support for multi function

address commits and fix lint

fix testcases and using a set to avoid duplicate func name

fix lint
minor linting

add dynamic support for tensorlist ops

add infering static shape for tensorlist ops using tensorliststack

locate tensorlist allocation node from write node in the subfunctions

fix a bug in tensorlist reserve

add testcase

fix lint

fix lint

fix lint

enable sequential test cases

Co-authored-by: David Huang <davhuan@amazon.com>
Co-authored-by: Rohan Mukherjee <mukrohan@amazon.com>
Co-authored-by: Xingyu Zhou <zhoxingy@amazon.com>
Co-authored-by: Srinidhi Goud <srinidhi.goud29@gmail.com>
Co-authored-by: Xiao <weix@amazon.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant