6f4c28e48a80d9774e2dd64bc442fc1ca528247f,src/graph_builder/graph/operators/deconvolution2d.py,Deconvolution2D,__call__,#Deconvolution2D#Any#Any#,32

Before Change

        W2 = (x_shape_dict[Axis.W] - 1) * self.SW - 2 * self.PW + self.KW
        C2 = w_shape_dict[Axis.N]

        if x.axis_order == OrderNCHW:
            var_shape = [N, C2, H2, W2]
        elif x.axis_order == OrderNHWC:
            var_shape = [N, H2, W2, C2]
        else:
            raise NotImplementedError()
        y = Variable(var_shape, x.axis_order)
        self.append_input("x", x)
        self.append_input("w", w)
        self.append_output("y", y)
        return y,

After Change

        W2 = (x_shape_dict[Axis.W] - 1) * self.SW - 2 * self.PW + self.KW
        C2 = w_shape_dict[Axis.N]

        y = Variable([N, C2, H2, W2], OrderNCHW)
        y.change_axis_order(x.axis_order)

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

Instances