AttributeError: module ‘tvm.relay’ has no attribute ‘ir_pass’
from tvm import relay def make_dataset(graph, size=100): args = relay.ir_pass.infer_type(graph).params def create_arr(var): ttype = var.type_annotation np_arr = np.random.uniform(-1.0, 1.0, size=ttype.concrete_shape).astype(ttype.dtype) return tvm.ndarray.array(np_arr) n, c, h, w = 1, 3, 224, 224 def make_graph(data): weight = relay.var(“conv_weight”) out = relay.nn.conv2d(data, weight, kernel_size=(3, 3), padding=(1, 1), channels=c) out = relay.Function(relay.ir_pass.free_vars(out), out)
Anyone can solve this problem?