strategy.register_specialized_implement(wrap_compute_conv2d(topi.x86.conv2d_winograd), topi.x86.conv2d_winograd, nv2d(topi.x86.conv2d_nchw), topi.x86.schedule_conv2d_nchw) elif layout == "NHWC": strategy.register_default_implement(wrap_compute_conv2d(topi.nn.conv2d_nhwc) topi.x86.schedule_conv2d_nhwc) elif layout == "NCHWc": strategy.register_default_implement(wrap_compute_conv2d(topi.nn.conv2d_nchwc), topi.x86.