lint

apache · Aug 3, 2021 · 4106158 · 4106158
1 parent 15e2d13
commit 4106158
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py
@@ -1452,7 +1452,9 @@ def linear(self, inputs, input_types):
         elif len(b_shape) == 1:
             mm_out = self.matmul([inputs[0], inputs[1]], input_types[:2])
         else:
-            mm_out = self.matmul([inputs[0], _op.transpose(inputs[1], axes=(1, 0))], input_types[:2])
+            mm_out = self.matmul(
+                [inputs[0], _op.transpose(inputs[1], axes=(1, 0))], input_types[:2]
+            )
         if isinstance(bias, _expr.Expr):
             bias_ndims = len(self.infer_shape_with_prelude(bias))
             if bias_ndims == 1: