b4910f5fc768b1304099295a8d8d2d8d08474469,src/graph_transpiler/webdnn/graph/operators/sgemm.py,Sgemm,__call__,#Sgemm#Any#Any#,30

Before Change

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

    def exec(self):
        A = self.inputs["A"]
        B = self.inputs["B"]

After Change

        if Placeholder.check_resolved(A.size) and Placeholder.check_resolved(self.M * self.K):
            assert A.size == self.M * self.K
        if Placeholder.check_resolved(B.size) and Placeholder.check_resolved(self.N * self.K):
            assert B.size == self.N * self.K

        C = Variable(self.parameters["out_shape"], self.parameters["out_order"])

        self.append_input("A", A)

Instances