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

Commits on Jul 15, 2021

  1. [BYOC, TRT] Update annotation pass to support multiple functions

    add support for multi function
    
    address commits and fix lint
    
    fix testcases and using a set to avoid duplicate func name
    
    fix lint
    Xingyu Zhou authored and rohanmukh committed Jul 15, 2021
    Configuration menu
    Copy the full SHA
    1d11de3 View commit details
    Browse the repository at this point in the history
  2. [Frontend, TF2] Added support for TensorList ops

    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>
    5 people committed Jul 15, 2021
    Configuration menu
    Copy the full SHA
    80c711d View commit details
    Browse the repository at this point in the history
  3. [Frontend, Prelude] Enable static batch size in Prelude for TensorLis…

    …t Stack outputs
    Xingyu Zhou authored and rohanmukh committed Jul 15, 2021
    Configuration menu
    Copy the full SHA
    cb285f0 View commit details
    Browse the repository at this point in the history