b4910f5fc768b1304099295a8d8d2d8d08474469,src/graph_transpiler/webdnn/graph/operators/col2im.py,Col2Im,__call__,#Col2Im#Any#,19

Before Change

    def __call__(self, col: Variable):
        self.append_input("col", col)
        return self.exec()

    def exec(self):
        col = self.inputs["col"]
        N = col.shape_dict[Axis.N]

After Change

        self.parameters["padding"] = to_tuple(padding)
        self.attributes.add(Tensorwise(Axis.N))

    def __call__(self, col: Variable):
        N = col.shape_dict[Axis.N]
        H2 = (col.shape_dict[Axis.H] - 1) * self.SH - 2 * self.PH + self.KH
        W2 = (col.shape_dict[Axis.W] - 1) * self.SW - 2 * self.PW + self.KW
        C2 = col.shape_dict[Axis.C]

        im = Variable([N, H2, W2, C2], OrderNHWC)

        self.append_input("col", col)
        self.append_output("im", im)

        return im,

    @property
    def ksize(self) -> Tuple[int, int]:
        return self.parameters["ksize"]

Instances