Merge pull request apache#12 from heliqi/paddle_frontend

black formatting
wjj19950828 · Aug 18, 2021 · 2a82fdb · 2a82fdb
2 parents c2d4bf5 + be0e04e
commit 2a82fdb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/tvm/relay/frontend/paddlepaddle.py b/python/tvm/relay/frontend/paddlepaddle.py
@@ -346,8 +346,9 @@ def convert_layer_norm(g, op, block):
     scale_input = op.input("Scale")
 
     x_shape = infer_shape(x)
-    assert (
-        begin_norm_axis in (len(x_shape) - 1, -1)
+    assert begin_norm_axis in (
+        len(x_shape) - 1,
+        -1,
     ), "Support only normalization over last one dimension."
 
     if bias_input: