@autotvm.register_topi_schedule('conv2d_nhwc_tensorcore.cuda')
def schedule_conv2d_nhwc_tensorcore(cfg, outs):
    'TOPI schedule callback'
    s = te.create_schedule([x.op for x in outs])

    def _callback(op):
        if ('conv2d_nhwc_tensorcore' in op.tag):
            schedule_nhwc_tensorcore_cuda(cfg, s, op.output(0))
    traverse_inline(s, outs[0].op, _callback)
    return s
TOPI schedule callback
@autotvm.register_topi_schedule('conv2d_nhwc_tensorcore.cuda')
def schedule_conv2d_nhwc_tensorcore(cfg, outs):
    
    s = te.create_schedule([x.op for x in outs])

    def _callback(op):
        if ('conv2d_nhwc_tensorcore' in op.tag):
            schedule_nhwc_tensorcore_cuda(cfg, s, op.output(0))
    traverse_inline(s, outs[0].op, _callback)
    return s