[GitHub] [tvm] zxy844288792 opened a new pull request #7502: [Frontend][Tensorflow] Support range like axis in tf.raw_ops.All for TF 2.x

[GitBox <gi...@apache.org>]

zxy844288792 opened a new pull request #7502:
URL: https://github.com/apache/tvm/pull/7502


   I tried to compile tf2 mobilev2_ssd model from tf hub model zoo. Instead of seeing tf.math.reduce_all, there are several tf.raw_ops.All in the graph_def and the axis is a range node instead of some constant. Hence, I added support for if all attributes(start, limit, delta) are constant in a range node, evaluate it directly by genearting corresponging relay.const.
   
   With this, I was able to compile and run inference successfully for tf2 mobilev2_ssd model.


----------------------------------------------------------------
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

[GitHub] [tvm] zhiics merged pull request #7502: [Frontend][Tensorflow] Support range like axis in tf.raw_ops.All for TF 2.x

[GitBox <gi...@apache.org>]

zhiics merged pull request #7502:
URL: https://github.com/apache/tvm/pull/7502


   


----------------------------------------------------------------
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