b4910f5fc768b1304099295a8d8d2d8d08474469,src/graph_transpiler/webdnn/graph/operators/zero_padding_2d.py,ZeroPadding2D,__call__,#ZeroPadding2D#Any#,37

Before Change

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

    def exec(self):
        x = self.inputs["x"]
        x_shape_dict = x.shape_dict

After Change

        self.attributes.add(Tensorwise(Axis.N))

    def __call__(self, x: Variable):
        x_shape_dict = x.shape_dict

        N = x_shape_dict[Axis.N]
        H2 = x_shape_dict[Axis.H] + 2 * self.parameters["padding"][0]
        W2 = x_shape_dict[Axis.W] + 2 * self.parameters["padding"][1]
        C2 = x_shape_dict[Axis.C]

        y = Variable([N, H2, W2, C2], OrderNHWC)
        y.change_order(x.order)  // output same order as input to preserve following reshape semantics

        self.append_input("x", x)
        self.append_output("y", y)
        return y,

Instances