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))
//