zxy844288792 opened a new pull request #8558: URL: https://github.com/apache/tvm/pull/8558
In current unpack implementation: https://github.com/apache/tvm/blob/9c63f4fc318652f6fff68342da2d11b26592a3e0/python/tvm/relay/frontend/tensorflow_ops.py#L2557 Any() is not supported and in some tf2 models from tf model hub like ssd_mobilenet_v2, we encounter some cases when unpack is after a tensorlist stack node. And the output for tensorlist stack will have any() in its dimension 0: https://github.com/apache/tvm/blob/9c63f4fc318652f6fff68342da2d11b26592a3e0/python/tvm/relay/prelude.py#L577. In this case it will not supported. This pr adds support by introducing an special case handling, when we know the tensorlist stack node input only contains one element, it will have output shape with 1 in dimension 0. -- 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. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
