How to get the conv shape in the function:schedule_conv2d

Questions
1

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