quic-sanirudh commented on code in PR #15258:
URL: https://github.com/apache/tvm/pull/15258#discussion_r1255629596
##########
python/tvm/contrib/hexagon/transform.py:
##########
@@ -311,3 +318,73 @@ def remove_empty_pad(mod):
"""Remove the empty pad operator."""
mod["main"] = rewrite(remove_empty_pad_callback(), mod["main"])
return mod
+
+
+class simplify_qnn_concat_in_func(DFPatternCallback):
+
+ """
+ Propagate qnn.concat's quantization params to its inputs,
+ and try to avoid redundant requantization while doing so.
+ """
Review Comment:
Minor comment. If possible we could add a snippet of the relay IR before and
after this pass in the comments similar to the comments mentioned in the
functions above (like `remove_empty_pad_callback` or `qdistilbert_rewrite`).
--
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]
