f6276d38368f18f39a72ee0cd121d9bdb0a1730c,src/graph_transpiler/webdnn/backend/webgl/kernels/im2col.py,,im2col,#Any#,14

Before Change

    C1 = im.shape_dict[Axis.C]

    assert im.order.check_same_axes(OrderNHWC)
    assert col.order.check_same_axes(OrderNHWC)
    assert ChannelMode.get(im) == ChannelModeEnum.R

    if ChannelMode.get(col) == ChannelModeEnum.R:
        code = KernelCode([

After Change

    col_shape = col.shape[0:3] + (mul(col.shape[3:6]),)
    col_stride = [mul(col_shape[i + 1:]) for i in range(len(col_shape))]
    col_order = Order(col.order.axes[0:3] + (Axis.C,))

    if ChannelMode.get(col) == ChannelModeEnum.R:
        code = KernelCode([
void main() {

Instances