diff --git a/python/tvm/relay/op/contrib/ethosu.py b/python/tvm/relay/op/contrib/ethosu.py index e53be02b8dd2..3016c3be4ce4 100644 --- a/python/tvm/relay/op/contrib/ethosu.py +++ b/python/tvm/relay/op/contrib/ethosu.py @@ -956,6 +956,8 @@ class MeanParams: composite_name = "ethos-u.mean" def __init__(self, func_body: Call): + from tvm.relay.backend.contrib.ethosu.util import RequantArgs + requantize = func_body mean_op = requantize.args[0] attrs = mean_op.attrs