tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [incubator-tvm] kevinthesun opened a new pull request #5794: [Frontend][Tensorflow]Improve TF Parser to keep output nodes for saved_model
Date Fri, 12 Jun 2020 22:12:59 GMT

kevinthesun opened a new pull request #5794:
URL: https://github.com/apache/incubator-tvm/pull/5794


   This PR fixes two things for some tf models in the format of saved model:
   1. Make fill not depend on node attr ```_output_shapes``` since we can always get that
by trying to ```_infer_value```. Some saved model will generate unknown rank shape for this
attr which is actually not true.
   2. Add output nodes as ```protected_nodes``` when parsing saved models.
   
   @yongwww @lixiaoquan @zhiics 
   


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org



Mime
View raw message