f6276d38368f18f39a72ee0cd121d9bdb0a1730c,test/webdnn_test/graph_test/operators_test/col2im_test.py,,,#,7

Before Change

    OrderHWCN
from webdnn.graph.variable import Variable

orders4 = [OrderNHWC, OrderHWNC, OrderHWCN, OrderNCHW, OrderCNHW, OrderCHWN]


def main(k, s, p, n, h1, w1, c1, expected_shape_dict: AxisKeyDict[int]):
    for order_x in orders4:

After Change

from webdnn.graph.order import Order, OrderNHWC
from webdnn.graph.variable import Variable

OrderNHWKKC = Order([Axis.N, Axis.H, Axis.W, Axis.KH, Axis.KW, Axis.C])


def main(col_shape=[1, 5, 5, 3, 3, 6], col_order=OrderNHWKKC, ksize=3, stride=1, padding=1,
         expected_shape_dict: AxisKeyDict[int] = AxisKeyDict(OrderNHWC.axes, [1, 5, 5, 6])):

Instances