from tvm import relay
from tvm import nd
import numpy as np
import torch
import torchvision
from op_runtime import op_run
from PIL import Image
from torchvision import transforms
from torchvision import models
import onnx
import tvm
onnx_model = onnx.load_model(‘vgg16.onnx’)
x = np.random.rand(1, 3, 224, 224)
input_name = ‘input0’
shape_dict = {input_name: x.shape}
mod, params = relay.frontend.from_onnx(onnx_model, shape_dict)