@autotvm.register_topi_compute('conv2d_nhwc_tensorcore.cuda')
def conv2d_nhwc_tensorcore(cfg, data, kernel, strides, padding, dilation, out_dtype):
    'Compute conv2d with tensorcore for NCHW layout'
    return nhwc_tensorcore_cuda(cfg, data, kernel, strides, padding, dilation, out_dtype)
Compute conv2d with tensorcore for NCHW layout
@autotvm.register_topi_compute('conv2d_nhwc_tensorcore.cuda')
def conv2d_nhwc_tensorcore(cfg, data, kernel, strides, padding, dilation, out_dtype):
    
    return nhwc_tensorcore_cuda(cfg, data, kernel, strides, padding, dilation, out_dtype)