[TE compiler] Check failed: (call_node->op.as<OpNode>()) is false: Primitive function only allows call into primitive ops

I want use te compiler to get the flops, but met this error(Check failed: (call_node->op.as()) is false: Primitive function only allows call into primitive ops)

my code is here //

onnx_model = onnx.load("/home/super_resolution.onnx")

mod, params = relay.frontend.from_onnx(onnx_model, shape_dict)

with tvm.transform.PassContext(opt_level=3):

opt_mode, _ = relay.optimize(mod, target, params)

tec = relay.backend.te_compiler.get()

lower_module = tec.lower(opt_mode[“main”], “llvm”)

flops = estimate_tir_flops(IRModule(lower_module))

//