From f09081b1e4684101b32a86953f22799a5f00f100 Mon Sep 17 00:00:00 2001 From: lhutton1 <35535092+lhutton1@users.noreply.github.com> Date: Thu, 2 Dec 2021 17:48:28 +0000 Subject: [PATCH] [microNPU] Add RequantizeArgs import to MeanParams (#9636) Fixes failing CI since CI on #9576 passed before #9624 was merged. Change-Id: I44f6aa0abe09d206f3e948e36e9799eaa12b3d14 --- python/tvm/relay/op/contrib/ethosu.py | 2 ++ 1 file changed, 2 insertions(+) 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