b4910f5fc768b1304099295a8d8d2d8d08474469,src/graph_transpiler/webdnn/graph/operators/reduce.py,Reduce,__call__,#Reduce#Any#,34

Before Change

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

    def exec(self):
        reduced_axis = self.axis

After Change

    def __call__(self, x: "variable.Variable"):
        reduced_axis = self.axis

        y_axes = list(x.order.axes)
        y_shape = [1 if axis == reduced_axis else x.shape_dict[axis] for axis in y_axes]
        y_order = Order(y_axes)

        y = variable.Variable(y_shape, y_order)

        for axis in x.order.axes:
            if axis != reduced_axis:
                self.attributes.add(Tensorwise(axis))

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

Instances