def _callback(op):
    if ('conv2d_nhwc_tensorcore' in op.tag):
        schedule_nhwc_tensorcore_cuda(cfg, s, op.output(0))
def _callback(op):
    if ('conv2d_nhwc_tensorcore' in op.tag):
        schedule_nhwc_tensorcore_cuda(cfg, s, op.output(0))