mshr-h commented on code in PR #18524:
URL: https://github.com/apache/tvm/pull/18524#discussion_r2572976194
##########
python/tvm/relax/transform/legalize_ops/statistical.py:
##########
@@ -18,14 +18,24 @@
"""Default legalization function for statistical operators."""
from typing import List
from tvm import topi, tir, te
+from tvm import relax
from ...block_builder import BlockBuilder
from ...expr import Call, Expr
from .common import TEFunc, LegalizeFunc, register_legalize
def _statistical(te_func: TEFunc) -> LegalizeFunc:
def statistical_call_te(bb: BlockBuilder, call: Call) -> Expr:
- return bb.call_te(te_func, call.args[0], call.attrs.axis,
call.attrs.keepdims)
+ arg = call.args[0]
Review Comment:
Should be handled in the frontend instead of legalization pass.
--
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]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
