Hi!
I am modifying the VTA schedule via editing the vta_conv2d.py. And how to get the conv shape (kernel, stride, pad, in_filters, out_filters) in the function schedule_conv2d while not changing the input and output params of this function?
@autotvm.register_topi_schedule(topi.generic.schedule_conv2d_nchw, 'vta', 'direct')
def _schedule_conv2d(cfg, outs):
assert len(outs) == 1
output = outs[0]
const_ops = []
ewise_inputs = []
ewise_ops = []
conv2d_res = []
assert "int" in output.op.input_tensors[0].dtype